Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -9,7 +9,7 @@ import numpy as np
|
|
| 9 |
import onnxruntime as rt
|
| 10 |
import pandas as pd
|
| 11 |
from PIL import Image
|
| 12 |
-
|
| 13 |
|
| 14 |
|
| 15 |
|
|
@@ -174,7 +174,7 @@ class Predictor:
|
|
| 174 |
|
| 175 |
return np.expand_dims(image_array, axis=0)
|
| 176 |
|
| 177 |
-
|
| 178 |
def predict(
|
| 179 |
self,
|
| 180 |
image,
|
|
|
|
| 9 |
import onnxruntime as rt
|
| 10 |
import pandas as pd
|
| 11 |
from PIL import Image
|
| 12 |
+
import spaces
|
| 13 |
|
| 14 |
|
| 15 |
|
|
|
|
| 174 |
|
| 175 |
return np.expand_dims(image_array, axis=0)
|
| 176 |
|
| 177 |
+
@spaces.GPU()
|
| 178 |
def predict(
|
| 179 |
self,
|
| 180 |
image,
|