Update app.py

app.py

@@ -1,31 +1,14 @@
 import gradio as gr
+import torch
 from rembg import remove
 from PIL import Image
-from flask import Flask, Response
-from flask_cors import CORS
+import io
 
-# Create a Flask app
-app = Flask(__name__)
-
-# Enable CORS for the Flask app
-CORS(app)
-
-# Your background removal function
 def remove_bg(image):
     output_image = remove(image)
     return output_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
-    )
-
-# Mount the Gradio app at the root path
-app = gr.mount_gradio_app(app, demo, path="/")
+demo = gr.Interface(fn=remove_bg, inputs=gr.Image(type="pil"), outputs="image")
 
-# Run the Flask app
 if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=7860)
+    demo.launch()