Spaces:
Runtime error
Runtime error
Harshad Bhandwaldar
commited on
Commit
•
d498fcd
1
Parent(s):
871fb7e
added
Browse files
app.py
CHANGED
|
@@ -16,8 +16,8 @@ model = gluoncv.model_zoo.get_model("psp_resnet101_ade", pretrained=True)
|
|
| 16 |
|
| 17 |
def segmentation(img):
|
| 18 |
img = Image.open(img)
|
| 19 |
-
img.load()
|
| 20 |
-
img = np.asarray(img, dtype="int32")
|
| 21 |
img = mx.ndarray.array(img)
|
| 22 |
ctx = mx.cpu(0)
|
| 23 |
img = test_transform(img, ctx)
|
|
|
|
| 16 |
|
| 17 |
def segmentation(img):
|
| 18 |
img = Image.open(img)
|
| 19 |
+
# img.load()
|
| 20 |
+
# img = np.asarray(img, dtype="int32")
|
| 21 |
img = mx.ndarray.array(img)
|
| 22 |
ctx = mx.cpu(0)
|
| 23 |
img = test_transform(img, ctx)
|