Update Dockerfile

Dockerfile

@@ -1,24 +1,28 @@
-# Use Python 3.12 as the base image
-FROM python:3.12-slim
+# Use Python 3.10 as the base image
+FROM python:3.10-slim
 
 # Install system dependencies
 RUN apt-get update && apt-get install -y \
     ffmpeg \
     git \
     build-essential \
+    ninja-build \
     && rm -rf /var/lib/apt/lists/*
 
 # Create a non-root user
 RUN useradd -m -u 1000 user
 WORKDIR /app
 
-# Install Python dependencies in stages
+# Install Python dependencies in correct order
 RUN pip install --no-cache-dir --upgrade pip && \
     # Install PyTorch first
     pip install --no-cache-dir \
-        torch \
-        torchvision && \
-    # Then install other dependencies
+        torch==2.1.0 \
+        torchvision==0.16.0 \
+        ninja && \
+    # Install AutoAWQ
+    pip install --no-cache-dir autoawq && \
+    # Install other dependencies
     pip install --no-cache-dir \
         git+https://github.com/huggingface/transformers \
         accelerate \
@@ -27,12 +31,7 @@ RUN pip install --no-cache-dir --upgrade pip && \
         uvicorn[standard] \
         python-multipart \
         pillow \
-        pydantic && \
-    # Finally install autoawq
-    # Install autoawq from wheel
-    pip3 install --no-cache-dir \
-        --find-links https://github.com/casper-hansen/AutoAWQ/releases/download/v0.2.8/ \
-        autoawq==0.2.8
+        pydantic
 
 # Copy application files
 COPY --chown=user:user . /app