Update Dockerfile

Dockerfile

@@ -12,20 +12,24 @@ RUN apt-get update && apt-get install -y \
 RUN useradd -m -u 1000 user
 WORKDIR /app
 
-# Install Python dependencies
+# Install Python dependencies in stages
 RUN pip install --no-cache-dir --upgrade pip && \
+    # Install PyTorch first
     pip install --no-cache-dir \
         torch \
-        torchvision \
+        torchvision && \
+    # Then install other dependencies
+    pip install --no-cache-dir \
         git+https://github.com/huggingface/transformers \
         accelerate \
         qwen-vl-utils[decord]==0.0.8 \
         fastapi \
         uvicorn[standard] \
         python-multipart \
-        autoawq \
         pillow \
-        pydantic
+        pydantic && \
+    # Finally install autoawq
+    pip install --no-cache-dir autoawq
 
 # Copy application files
 COPY --chown=user:user . /app