Update Dockerfile

Dockerfile

@@ -55,18 +55,27 @@ RUN pip install --no-cache-dir -r requirements.txt
 COPY . .
 
 # ---------------------------------------------------------------------------
-# Install BasicSR (required for CodeFormer)
-# Must cd into CodeFormer so setup.py finds ./basicsr/VERSION correctly
+# Install realesrgan (pulls PyPI basicsr==1.4.2 as dependency)
 # ---------------------------------------------------------------------------
-RUN cd /app/CodeFormer && python basicsr/setup.py develop 2>&1 || true
+RUN pip install --no-cache-dir realesrgan 2>&1 || true
 
 # ---------------------------------------------------------------------------
-# Install realesrgan
+# FIX: Uninstall broken PyPI basicsr (it imports removed
+# torchvision.transforms.functional_tensor). CodeFormer's own local
+# basicsr (at /app/CodeFormer/basicsr/) works fine and is used instead
+# via PYTHONPATH below.
 # ---------------------------------------------------------------------------
-RUN pip install --no-cache-dir realesrgan 2>&1 || true
+RUN pip uninstall -y basicsr 2>&1 || true
+
+# ---------------------------------------------------------------------------
+# Use CodeFormer's local basicsr + facelib via PYTHONPATH
+# This makes `import basicsr` and `import facelib` resolve to the
+# CodeFormer-bundled versions that are compatible with modern torchvision.
+# ---------------------------------------------------------------------------
+ENV PYTHONPATH="/app/CodeFormer:${PYTHONPATH}"
 
 # ---------------------------------------------------------------------------
-# Download pretrained models
+# Download pretrained models (now uses local basicsr — no import errors)
 # ---------------------------------------------------------------------------
 RUN python CodeFormer/scripts/download_pretrained_models.py facelib 2>&1 || true
 RUN python CodeFormer/scripts/download_pretrained_models.py CodeFormer 2>&1 || true
@@ -89,6 +98,10 @@ EXPOSE 7860
 # ---------------------------------------------------------------------------
 CMD ["gunicorn","app:fastapi_app","--workers=1","--worker-class=uvicorn.workers.UvicornWorker","--bind=0.0.0.0:7860","--timeout=600","--graceful-timeout=300","--keep-alive=5","--max-requests=200","--max-requests-jitter=50","--preload","--access-logfile=-","--error-logfile=-","--log-level=info"]
 
+
+
+
+
 # # ============================================================================
 # # FaceSwap API - HuggingFace Spaces Docker SDK (FIXED)
 # # Production-ready with Gunicorn + Uvicorn workers