Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 from rembg import remove
-from fastapi import FastAPI
-from fastapi.staticfiles import StaticFiles
+from fastapi import FastAPI, UploadFile, File, HTTPException
+from fastapi.responses import Response
 import os
 
 # Create FastAPI app
@@ -32,48 +32,10 @@ app.mount("/", gradio_app)
 
 # API endpoint
 @app.post("/api/remove-bg")
-async def api_remove_bg(file: bytes = File(...)):
+async def api_remove_bg(file: UploadFile = File(...)):
     try:
-        output_bytes = remove(file)
-        return Response(content=output_bytes, media_type="image/png")
-    except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))import gradio as gr
-from rembg import remove
-from fastapi import FastAPI
-from fastapi.staticfiles import StaticFiles
-import os
-
-# Create FastAPI app
-app = FastAPI()
-
-# Background removal function
-def remove_bg(image_path):
-    with open(image_path, "rb") as f:
-        input_image = f.read()
-    output_image = remove(input_image)
-    output_path = "/tmp/output.png"
-    with open(output_path, "wb") as f:
-        f.write(output_image)
-    return output_path
-
-# Create Gradio interface
-gradio_app = gr.Interface(
-    fn=remove_bg,
-    inputs=gr.Image(type="filepath", label="Input Image"),
-    outputs=gr.Image(type="filepath", label="Output Image"),
-    title="Background Remover",
-    description="Upload an image to remove the background",
-    allow_flagging="never"
-)
-
-# Mount Gradio app
-app.mount("/", gradio_app)
-
-# API endpoint
-@app.post("/api/remove-bg")
-async def api_remove_bg(file: bytes = File(...)):
-    try:
-        output_bytes = remove(file)
+        input_image = await file.read()
+        output_bytes = remove(input_image)
         return Response(content=output_bytes, media_type="image/png")
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))