Update app.py

app.py

@@ -33,10 +33,17 @@ predictor = DefaultPredictor(cfg)
 
 def segment_buildings(im):
 
+    # Convert PIL Image to NumPy array
+    im = np.array(im)
+
     outputs = predictor(im)
+
+    # We can use `Visualizer` to draw the predictions on the image.
     v = Visualizer(im[:, :, ::-1], MetadataCatalog.get(cfg.DATASETS.TRAIN[0]), scale=1.2)
-    out = v.draw_instance_predictions(outputs["instances"].to("cpu"))
-    out_im = out.get_image()[:, :, ::-1]
+    out_im = v.draw_instance_predictions(outputs["instances"].to("cpu")).get_image()
+
+    # Convert the output image back to PIL Image
+    out_im = Image.fromarray(out_im)
     return out_im
 
 # gradio components