Upload 4 files

Dockerfile

@@ -0,0 +1,19 @@
+FROM python:3.10-slim
+
+WORKDIR /app
+
+RUN apt-get update && apt-get install -y --no-install-recommends libgl1 libglib2.0-0 \
+    && rm -rf /var/lib/apt/lists/*
+
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Descarga el modelo de rembg durante el build, para que el primer
+# pedido de un usuario no tenga que esperar la descarga.
+RUN python -c "import rembg; rembg.new_session('u2net')"
+
+COPY app.py .
+
+EXPOSE 7860
+
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

README.md

@@ -1,10 +1,24 @@
 ---
-title: Remove Bg Api
-emoji: 📊
-colorFrom: purple
-colorTo: yellow
+title: Remove Background API
+emoji: ✂️
+colorFrom: green
+colorTo: gray
 sdk: docker
+app_port: 7860
 pinned: false
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+# Remove Background API
+
+Servidor simple que quita el fondo de imágenes usando [rembg](https://github.com/danielgatis/rembg).
+Gratis, sin límite de imágenes, corre en el CPU gratuito de Hugging Face Spaces.
+
+## Endpoint
+
+`POST /remove-bg` — recibe un campo `file` (multipart/form-data) con la imagen,
+devuelve un PNG con fondo transparente.
+
+## Nota sobre el plan gratuito
+
+Los Spaces gratuitos se "duermen" tras un rato sin uso. El primer pedido después
+de estar dormido puede tardar 30–60 segundos en despertar; los siguientes son rápidos.

app.py

@@ -0,0 +1,34 @@
+from fastapi import FastAPI, File, UploadFile, HTTPException
+from fastapi.middleware.cors import CORSMiddleware
+from fastapi.responses import Response
+from rembg import remove
+
+app = FastAPI(title="Remove Background API")
+
+# Permite que tu página en GitHub Pages (u otro origen) llame a este servidor.
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+
+@app.get("/")
+def health():
+    return {"status": "ok"}
+
+
+@app.post("/remove-bg")
+async def remove_bg(file: UploadFile = File(...)):
+    if not file.content_type or not file.content_type.startswith("image/"):
+        raise HTTPException(status_code=400, detail="El archivo debe ser una imagen.")
+
+    input_bytes = await file.read()
+
+    try:
+        output_bytes = remove(input_bytes)
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"No se pudo procesar la imagen: {e}")
+
+    return Response(content=output_bytes, media_type="image/png")

requirements.txt

@@ -0,0 +1,6 @@
+fastapi==0.115.0
+uvicorn[standard]==0.30.6
+python-multipart==0.0.9
+rembg==2.0.59
+onnxruntime==1.19.2
+pillow==10.4.0