Update app.py

app.py

@@ -31,7 +31,7 @@ cfg.MODEL.WEIGHTS = "model_weights/chatswood_buildings_poc.pth"
 cfg.MODEL.ROI_HEADS.NUM_CLASSES = 8
 predictor = DefaultPredictor(cfg)
 
-def segment_buildings(im):
+def segment_buildings(inp):
 
     im = cv2.imread(im)
     outputs = predictor(im)
@@ -49,16 +49,16 @@ gr_slider_confidence = gr.inputs.Slider(0,1,.1,.7,
                                         label='Set confidence threshold % for masks')
 """
 # gradio outputs
-inputs = gr.inputs.Image(type="pil", label="Input Image")
-outputs = gr.outputs.Image(type="pil", label="Output Image")
+# inputs = gr.inputs.Image(type="pil", label="Input Image")
+# outputs = gr.outputs.Image(type="pil", label="Output Image")
 
 title = "Building Segmentation"
 description = "An instance segmentation demo for identifying boundaries of buildings in aerial images using DETR (End-to-End Object Detection) model with MaskRCNN-101 backbone"
 
 # Create user interface and launch
 gr.Interface(segment_buildings, 
-                inputs = inputs,
-                outputs = outputs,
+                inputs = "image",
+                outputs = "image",
                 title = title,
                 enable_queue = True,
                 description = description).launch()