Update app.py

app.py

@@ -35,9 +35,13 @@ def segment_buildings(im):
 
     im = cv2.imread(im)
     outputs = predictor(im)
-    v = Visualizer(im[:, :, ::-1], MetadataCatalog.get(cfg.DATASETS.TRAIN[0]), scale=1.2)
+    v = Visualizer(im[:, :, ::-1],
+                   scale=0.5,
+                   instance_mode=ColorMode.IMAGE_BW
+    )
     out = v.draw_instance_predictions(outputs["instances"].to("cpu"))
-    return Image.fromarray(np.uint8(out.get_image())).convert('RGB')
+    im = Image.fromarray(out.get_image()[:, :, ::-1])
+    return im
 
 # gradio components 
 """