updated app

app.py

@@ -26,26 +26,24 @@ def segment_sidewalk(image):
     save_image(prediction, 'mask.png', cmap='gray')
     return Image.open('image.png'), Image.open('mask.png'), Image.open('prob.png')
 
-def save_image(image, path, **kwargs):
-    plt.figure(figsize=(8, 8))
-    plt.imshow(image, interpolation='nearest', **kwargs)
-    plt.axis('off')
-    plt.tight_layout()
-    plt.savefig(path, bbox_inches='tight', pad_inches=0)
-    plt.close()
+# def save_image(image, path, **kwargs):
+#     plt.figure(figsize=(8, 8))
+#     plt.imshow(image, interpolation='nearest', **kwargs)
+#     plt.axis('off')
+#     plt.tight_layout()
+#     plt.savefig(path, bbox_inches='tight', pad_inches=0)
+#     plt.close()
 
 with gr.Blocks() as demo:
     with gr.Row():
         with gr.Column():
             image_input = gr.Image(type='pil', label='TIFF Image')
-            # threshold_slider = gr.Slider(minimum=0, maximum=1, step=0.01, value=0.5, label='Prediction Threshold')
-            segment_button = gr.Button('Get Sidewalk Mask')
+            button = gr.Button('Get Sidewalk Mask')
         with gr.Column():
             mask = gr.Image(type='pil', label='Predicted Mask')
             prob_map = gr.Image(type='pil', label='Predicted Probability Map')
-    segment_button.click(
-        segment_sidewalk, 
-        # inputs=[image_input, threshold_slider], 
+    button.click(
+        segment_sidewalk,  
         inputs=[image_input], 
         outputs=[image_input, mask, prob_map]
     )