Update app.py

app.py

@@ -1,30 +1,23 @@
-from fastapi import FastAPI, File, UploadFile, HTTPException, Header
-from fastapi.responses import StreamingResponse
+import gradio as gr
 from rembg import remove
 from PIL import Image
 import io
 
-app = FastAPI()
-
-# Replace this with your secret
-API_KEY = "your_super_secret_key"
-
-@app.post("/remove-bg")
-async def remove_bg(file: UploadFile = File(...), authorization: str = Header(None)):
-    # Check API key in Authorization header
-    if authorization != f"Bearer {API_KEY}":
-        raise HTTPException(status_code=401, detail="Unauthorized")
-
-    # Read image
-    contents = await file.read()
-    input_image = Image.open(io.BytesIO(contents))
-
-    # Remove background
+def remove_background(input_image):
+    if isinstance(input_image, str):
+        input_image = Image.open(input_image)
     output_image = remove(input_image)
+    return output_image
 
-    # Prepare image response
-    img_byte_arr = io.BytesIO()
-    output_image.save(img_byte_arr, format="PNG")
-    img_byte_arr.seek(0)
+demo = gr.Interface(
+    fn=remove_background,
+    inputs=gr.Image(type="pil"),
+    outputs=gr.Image(type="pil"),
+    title="Background Remover",
+    description="Upload an image to remove its background",
+    examples=[],
+    theme=gr.themes.Soft()
+)
 
-    return StreamingResponse(img_byte_arr, media_type="image/png")
+if __name__ == "__main__":
+    demo.launch()