Spaces:

adpro
/

semantic-08

Sleeping

adpro commited on Feb 8, 2023

Commit

1fc57d5

1 Parent(s): d65768b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,7 @@ from gluoncv.data.transforms.presets.segmentation import test_transform
 # using cpu
 ctx = mx.cpu(0)
-model = gluoncv.model_zoo.get_model("psp_resnet101_ade", pretrained=True)
 def segmentation(image):

 # using cpu
 ctx = mx.cpu(0)
+model = gluoncv.model_zoo.get_model("deeplab_resnet101_ade", pretrained=True)
 def segmentation(image):