Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -45,7 +45,7 @@ def preprocess(img):
|
|
| 45 |
img = transform_fn(img)
|
| 46 |
img = img.expand_dims(axis=0) # batchify
|
| 47 |
|
| 48 |
-
return img
|
| 49 |
|
| 50 |
|
| 51 |
def predict(path):
|
|
|
|
| 45 |
img = transform_fn(img)
|
| 46 |
img = img.expand_dims(axis=0) # batchify
|
| 47 |
|
| 48 |
+
return img.asnumpy()
|
| 49 |
|
| 50 |
|
| 51 |
def predict(path):
|