Add enhancer model

.gitignore

@@ -0,0 +1,25 @@
+# Python artifacts
+__pycache__/
+*.py[cod]
+*.env
+.venv/
+env/
+venv/
+
+# Data / model weights (you generally load these at runtime)
+weights/
+*.pth
+*.ckpt
+*.h5
+*.npz
+
+# IDE / editor configs
+.vscode/
+.idea/
+
+# OS files
+.DS_Store
+Thumbs.db
+
+# Hugging Face tokens (if you store any locally)
+*.hf_token

Dockerfile

@@ -0,0 +1,28 @@
+# 1. Base image
+FROM python:3.9
+
+# 2. Install system deps for RealESRGAN & PIL
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends \
+      git build-essential \
+      libsm6 libxext6 libxrender1 \
+      libgl1-mesa-glx && \
+    rm -rf /var/lib/apt/lists/*
+
+# 3. Create non-root user (for pip cache ownership)
+RUN useradd -m -u 1000 user
+USER user
+ENV PATH="/home/user/.local/bin:$PATH"
+
+# 4. Workdir & Python deps
+WORKDIR /app
+COPY --chown=user requirements.txt .
+RUN pip install --no-cache-dir --upgrade pip \
+ && pip install --no-cache-dir -r requirements.txt
+
+# 5. Copy app code
+COPY --chown=user . .
+
+# 6. Expose & run
+EXPOSE 7860
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860", "--log-level", "debug"]

app.py

@@ -0,0 +1,28 @@
+import io
+import torch
+
+from fastapi import FastAPI, File, UploadFile, HTTPException
+from fastapi.responses import StreamingResponse
+from PIL import Image
+from RealESRGAN import RealESRGAN
+
+app = FastAPI()
+
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+model = RealESRGAN(device, scale=4)
+model.load_weights("weights/RealESRGAN_x4.pth", download=True)
+
+
+@app.post("/enhance-img")
+async def enhance_img(file: UploadFile = File(...)):
+    try:
+        data = await file.read()
+        img = Image.open(io.BytesIO(data)).convert("RGB")
+        sr = model.predict(img)
+        buf = io.BytesIO()
+        sr.save(buf, format="PNG")
+        buf.seek(0)
+        return StreamingResponse(buf, media_type="image/png")
+
+    except Exception as e:
+        raise HTTPException(500, detail=str(e))

requirements.txt

@@ -0,0 +1,9 @@
+fastapi
+uvicorn[standard]
+torch>=1.7
+torchvision>=0.8.0
+Pillow
+numpy
+git+https://github.com/sberbank-ai/Real-ESRGAN.git
+huggingface_hub<0.13.0
+python-multipart