Update app.py

app.py

@@ -6,31 +6,27 @@ from fastapi.responses import Response
 from rembg import remove, new_session
 from PIL import Image
 
-# Model dimuat SEKALI saat startup agar request berikutnya cepat.
-# "u2netp" = model ringan (~4MB, download cepat). 
-# Ganti ke "u2net" atau "isnet-general-use" jika ingin kualitas lebih tinggi.
+# Model dimuat SEKALI saat startup
 session = new_session("u2netp")
 
 def remove_bg(img: Image.Image) -> Image.Image:
     return remove(img, session=session)
 
-# ---------- 1) UI GRADIO (sekalian jadi health-check Hugging Face) ----------
+# ---------- 1) UI GRADIO (sekaligus health-check) ----------
 demo = gr.Interface(
     fn=remove_bg,
     inputs=gr.Image(type="pil", label="Upload Gambar"),
     outputs=gr.Image(type="pil", label="Hasil (PNG Transparan)"),
     title="Remove Background API",
     description="Backend remove background. REST API: POST /remove-bg",
-    api_name="remove_bg",
 )
 
 # ---------- 2) REST API FASTAPI (untuk frontend Vercel) ----------
 app = FastAPI()
 
-# Izinkan frontend Vercel memanggil API ini (CORS)
 app.add_middleware(
     CORSMiddleware,
-    allow_origins=["*"],  # nanti bisa diganti domain vercel.app kamu
+    allow_origins=["*"],
     allow_methods=["*"],
     allow_headers=["*"],
 )
@@ -48,9 +44,7 @@ async def api_remove_bg(file: UploadFile = File(...)):
     output_img.save(buf, format="PNG")
     return Response(content=buf.getvalue(), media_type="image/png")
 
-# Gabungkan: UI Gradio di "/", REST API di /remove-bg
+# Gabungkan: Gradio UI di "/", REST API di /remove-bg
 app = gr.mount_gradio_app(app, demo, path="/")
 
-if __name__ == "__main__":
-    import uvicorn
-    uvicorn.run(app, host="0.0.0.0", port=7860)
+# JANGAN pakai uvicorn.run() — Hugging Face Spaces akan menjalankan `app` secara otomatis