Update app.py

app.py

@@ -48,7 +48,7 @@ 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(predict_building_mask, 
+gr.Interface(segment_buildings, 
                 inputs = gr_image_input,
                 outputs = gr_image_output,
                 title = title,