Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -9,7 +9,7 @@ from gluoncv.data.transforms.presets.segmentation import test_transform
|
|
9 |
# using cpu
|
10 |
ctx = mx.cpu(0)
|
11 |
|
12 |
-
model = gluoncv.model_zoo.get_model("
|
13 |
|
14 |
|
15 |
def segmentation(image):
|
|
|
9 |
# using cpu
|
10 |
ctx = mx.cpu(0)
|
11 |
|
12 |
+
model = gluoncv.model_zoo.get_model("deeplab_resnet101_ade", pretrained=True)
|
13 |
|
14 |
|
15 |
def segmentation(image):
|