Update app.py

app.py

@@ -3,12 +3,26 @@ import torch
 from rembg import remove
 from PIL import Image
 import io
+from fastapi.middleware.cors import CORSMiddleware
 
 def remove_bg(image):
     output_image = remove(image)
     return output_image
 
+# Gradio interface
 demo = gr.Interface(fn=remove_bg, inputs=gr.Image(type="pil"), outputs="image")
 
+# FastAPI app for adding CORS middleware
+app = demo.launch(prevent_thread_lock=True)
+
+# Adding CORS to allow requests from anywhere
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],  # Allows requests from any origin
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()