Fixed some bugs

app.py

@@ -63,9 +63,9 @@ def process_image(image_input):
 
     # Convert numpy arrays back to PIL Images for Gradio output
     seg_image = Image.fromarray(seg * 255)  # Convert boolean mask to uint8 image
-    prob_map = Image.fromarray((seg_prob * 255).astype(np.uint8))  # Scale probabilities to 0-255
+    # prob_map = Image.fromarray((seg_prob * 255).astype(np.uint8))  # Scale probabilities to 0-255
 
-    return image, seg_image, prob_map
+    return seg_image
 
 
 iface = gr.Interface(fn= process_image, inputs="image", outputs="image", title="Greeter")