Update app.py

app.py

@@ -50,18 +50,15 @@ def predict_segmentation(image):
     # Create an alpha channel: 255 where there is segmentation, 0 otherwise
     rgba_img[:, :, 3] = np.where(predicted_img_resized > 0, 255, 0)
 
-    # Convert the numpy array to an image and save it as a PNG file
+    # Convert the numpy array to an image and return it
     output_image = Image.fromarray(rgba_img)
-    output_image_path = "segmented_output.png"
-    output_image.save(output_image_path)
-
-    return output_image_path
+    return output_image
 
 # Gradio Interface
 iface = gr.Interface(
     fn=predict_segmentation,
     inputs="image",
-    outputs="file",
+    outputs="image",
     live=False
 )