Update app.py

app.py

@@ -39,8 +39,12 @@ def inference(image):
 
     img = np.array(image.resize((500, height)))
     outputs = predictor(img)
-
-    v = Visualizer(img,scale=1.2)
+    v = Visualizer(img[:, :, ::-1],
+                   #metadata=val_metadata_dicts, 
+                   scale=0.5, 
+                   instance_mode=ColorMode.IMAGE_BW   # remove the colors of unsegmented pixels. This option is only available for segmentation models
+    )
+    # v = Visualizer(img,scale=1.2)
     out = v.draw_instance_predictions(outputs["instances"].to("cpu"))
     
     return out.get_image()