Update app.py

app.py

@@ -56,7 +56,6 @@ def extract_image(pos_prompts, neg_prompts, img, threshold):
 
     return output_image, final_mask
 
-# Define Gradio interface
 # Define Gradio interface
 iface = gr.Interface(
     fn=extract_image,
@@ -69,18 +68,14 @@ iface = gr.Interface(
             label="Please describe what you want to ignore (comma separated)",
             key="neg_prompts",
         ),
-        gr.Image(type="pil", label="Input Image"),
-        gr.Slider(minimum=0, maximum=1, default=0.4, label="Threshold"),
+        gr.Image(type="pil", label="Input Image", key="img"),
+        gr.Slider(minimum=0, maximum=1, default=0.4, label="Threshold", key="threshold"),
     ],
     outputs=[
-        gr.Image(label="Result"),
-        gr.Image(label="Mask"),
+        gr.Image(label="Result", key="output_image"),
+        gr.Image(label="Mask", key="output_mask"),
     ],
 )
 
-# Launch Gradio API
-iface.launch()
-
-
-# Launch Gradio API
-iface.launch()
+# Launch Gradio API with an explicit endpoint
+iface.launch(share=True, debug=True, host="0.0.0.0", port=7860, open_browser=False)