Update app.py

app.py

@@ -1,24 +1,8 @@
-from fastapi import FastAPI, UploadFile, File, Response, HTTPException
-from rembg import remove
-from PIL import Image
-import io
-
-app = FastAPI(title="RemoveBG (rembg) – FastAPI")
-
-@app.get("/")
-def root():
-    return {"ok": True, "message": "Use POST /removebg with multipart field 'file'"}
-
-@app.post("/removebg")
-async def removebg(file: UploadFile = File(...)):
-    try:
-        # baca bytes gambar
-        b = await file.read()
-        # pastikan bisa dibuka PIL (validasi ringan)
-        Image.open(io.BytesIO(b))
-
-        # rembg menerima bytes atau PIL image
-        out_bytes = remove(b)  # hasil PNG RGBA
-        return Response(content=out_bytes, media_type="image/png")
-    except Exception as e:
-        raise HTTPException(status_code=400, detail=f"Failed to process: {e}")
+from gradio_client import Client, handle_file
+
+client = Client("not-lain/background-removal")
+result = client.predict(
+		f=handle_file('https://raw.githubusercontent.com/gradio-app/gradio/main/test/test_files/bus.png'),
+		api_name="/png"
+)
+print(result)