Update app.py

app.py

@@ -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("deeplab_resnet50_citys", pretrained=True)
+model = gluoncv.model_zoo.get_model("deeplab_resnet101_ade", pretrained=True)
 
 
 def segmentation(image):
@@ -18,7 +18,7 @@ def segmentation(image):
     img = test_transform(img, ctx)
     output = model.predict(img)
     predict = mx.nd.squeeze(mx.nd.argmax(output, 1)).asnumpy()
-    mask = get_color_pallete(predict, "cityscapes")
+    mask = get_color_pallete(predict, "ade20k")
     return mask