Update app.py

app.py

@@ -1,41 +1,17 @@
 import gradio as gr
 from rembg import remove
-from fastapi import FastAPI, UploadFile, File, HTTPException
-from fastapi.responses import Response
-import os
 
-# Create background removal function
-def remove_background(image_bytes):
-    try:
-        return remove(image_bytes)
-    except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
+def remove_background(input_image):
+    output = remove(input_image)
+    return output
 
-# Create Gradio interface
-def gradio_interface(input_image):
-    if input_image is None:
-        return None
-    return remove_background(input_image)
-
-# Create FastAPI app
-app = FastAPI()
-
-# Create and mount Gradio app
-gradio_app = gr.Interface(
-    fn=gradio_interface,
-    inputs=gr.Image(type="filepath"),
-    outputs=gr.Image(type="filepath"),
+interface = gr.Interface(
+    fn=remove_background,
+    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"
 )
 
-# API endpoint
-@app.post("/api/remove-bg")
-async def api_remove_bg(file: UploadFile = File(...)):
-    image_bytes = await file.read()
-    output_bytes = remove_background(image_bytes)
-    return Response(content=output_bytes, media_type="image/png")
-
-# Mount Gradio app
-app = gr.mount_gradio_app(app, gradio_app, path="/")
+interface.launch(server_name="0.0.0.0", server_port=7860)