Update app.py

app.py

@@ -1,10 +1,10 @@
-import os
-from rembg import remove
-from io import BytesIO
 from fastapi import FastAPI, UploadFile, File, HTTPException
 from fastapi.responses import StreamingResponse
+from rembg import remove
+from io import BytesIO
 import gradio as gr
 
+# Create FastAPI app
 app = FastAPI()
 
 # Background removal function
@@ -22,6 +22,15 @@ def gradio_interface(input_image):
     output_bytes = remove_background(input_image)
     return output_bytes
 
+# Create Gradio app
+gradio_app = gr.Interface(
+    fn=gradio_interface,
+    inputs=gr.Image(type="filepath", label="Input Image"),
+    outputs=gr.Image(type="numpy", label="Output Image"),
+    title="Background Remover",
+    description="Upload an image to remove the background"
+)
+
 # API endpoint
 @app.post("/api/remove-bg")
 async def api_remove_bg(file: UploadFile = File(...)):
@@ -33,14 +42,5 @@ async def api_remove_bg(file: UploadFile = File(...)):
     
     return StreamingResponse(BytesIO(output_bytes), media_type="image/png")
 
-# Gradio app
-gr_app = gr.Interface(
-    fn=gradio_interface,
-    inputs=gr.Image(type="filepath", label="Input Image"),
-    outputs=gr.Image(type="numpy", label="Output Image"),
-    title="Background Remover",
-    description="Upload an image to remove the background"
-)
-
-# Mount Gradio app
-app = gr.mount_gradio_app(app, gr_app, path="/")
+# Mount Gradio app to FastAPI
+app = gr.mount_gradio_app(app, gradio_app, path="/")