Update Dockerfile

Dockerfile

@@ -1,8 +1,8 @@
-FROM python:3.10-slim
+FROM python:3.9-slim
 
 WORKDIR /app
 
-# Fix: Use libgl1 instead of libgl1-mesa-glx for newer Debian
+# Install system dependencies
 RUN apt-get update && apt-get install -y \
     libgl1 \
     libglib2.0-0 \
@@ -10,13 +10,16 @@ RUN apt-get update && apt-get install -y \
     git \
     && rm -rf /var/lib/apt/lists/*
 
-# Install heavy dependencies separately to use cache
-RUN pip install --no-cache-dir torch torchvision --index-url https://download.pytorch.org/whl/cpu
-RUN pip install --no-cache-dir fastapi uvicorn python-multipart Pillow numpy opencv-python-headless
+# Install Torch and Torchvision versions that are compatible with basicsr
+RUN pip install --no-cache-dir torch==1.13.1 torchvision==0.14.1 --index-url https://download.pytorch.org/whl/cpu
 
-# Install Real-ESRGAN and its dependencies
+# Install other requirements
+RUN pip install --no-cache-dir fastapi uvicorn python-multipart Pillow numpy opencv-python-headless 
 RUN pip install --no-cache-dir basicsr realesrgan
 
+# 🔥 THE FIX: This command fixes the "functional_tensor" error inside the basicsr library
+RUN sed -i "s/from torchvision.transforms.functional_tensor import rgb_to_grayscale/from torchvision.transforms.functional import rgb_to_grayscale/" /usr/local/lib/python3.9/site-packages/basicsr/data/degradations.py
+
 COPY . .
 
 EXPOSE 7860