Update app.py

app.py

@@ -15,17 +15,16 @@ def remove_bg(image):
     output_image = remove(image)
     return output_image
 
-# Create Gradio interface
-demo = gr.Interface(
-    fn=remove_bg,
-    inputs=gr.Image(type="pil"),  # Accept PIL images
-    outputs="image",  # Output as an image
-)
+# Create Gradio interface using Blocks
+with gr.Blocks() as demo:
+    gr.Interface(
+        fn=remove_bg,
+        inputs=gr.Image(type="pil"),  # Accept PIL images
+        outputs="image",  # Output as an image
+    )
 
-# Wrap the Gradio app in the Flask app
-@app.route("/", methods=["GET", "POST"])
-def gradio_interface():
-    return demo.launch(share=True, server_name="0.0.0.0", server_port=7860)
+# Mount the Gradio app at the root path
+app = gr.mount_gradio_app(app, demo, path="/")
 
 # Run the Flask app
 if __name__ == "__main__":