fixed some bugs

app.py

@@ -67,13 +67,7 @@ def process_image(image_input):
 
     return image, seg_image, prob_map
 
-# Define Gradio interface
-iface = gr.Interface(
-    fn=process_image, 
-    inputs=gr.inputs.Image(shape=(256, 256)), 
-    outputs=[gr.outputs.Image(label="Original Image"), gr.outputs.Image(label="Segmentation Mask"), gr.outputs.Image(label="Probability Map")],
-    title="Image Segmentation"
-)
-
-# Launch the interface
+
+iface = gr.Interface(fn= process_image, inputs="image", outputs="image", title="Greeter")
+
 iface.launch()