Spaces:

Andy1621
/

uniformer_image_segmentation

Runtime error

SakuraD commited on May 23, 2022

Commit

31cc0ef

1 Parent(s): 272c94f

update

Files changed (1) hide show

app.py CHANGED Viewed

@@ -26,7 +26,7 @@ def set_example_image(example: list) -> dict:
 def inference(img):
     result = inference_segmentor(model, img)
-    res_img = show_result_pyplot(model, img, result, get_palette('cityscapes'))
     return res_img

 def inference(img):
     result = inference_segmentor(model, img)
+    res_img = show_result_pyplot(model, img, result, get_palette('ade'))
     return res_img