Upload main.py

from fastapi import FastAPI, UploadFile, File
from rembg import remove
from io import BytesIO
from starlette.responses import Response

app = FastAPI()



@app
.get("/")
def home():
return {"message": "Halo! API di Hugging Face siap digunakan."}



@app
.post("/hapus-background")
async def hapus_background(file: UploadFile = File(...)):
gambar_asli = await file.read()
gambar_bersih = remove(gambar_asli)
return Response(content=gambar_bersih, media_type="image/png")

main.py

@@ -0,0 +1,24 @@
+from fastapi import FastAPI, UploadFile, File
+from rembg import remove
+from io import BytesIO
+from starlette.responses import Response
+
+# Membuat aplikasi
+app = FastAPI()
+
+@app.get("/")
+def home():
+    return {"message": "Halo! API Penghapus Background siap digunakan."}
+
+# Ini fitur utamanya
+@app.post("/hapus-background")
+async def hapus_background(file: UploadFile = File(...)):
+    # 1. Baca gambar yang diupload user
+    gambar_asli = await file.read()
+    
+    # 2. Proses hapus background pakai rembg
+    gambar_bersih = remove(gambar_asli)
+    
+    # 3. Kembalikan hasilnya sebagai gambar PNG
+    return Response(content=gambar_bersih, media_type="image/png")
+