Update Dockerfile

Dockerfile

@@ -15,28 +15,29 @@ WORKDIR /app
 
 # Install Python dependencies in correct order
 RUN pip install --no-cache-dir --upgrade pip && \
-    # Install PyTorch first
+    # Install numpy<2 first for compatibility
+    pip install --no-cache-dir 'numpy<2' && \
+    # Install PyTorch and other core dependencies
     pip install --no-cache-dir \
         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 \
+        ninja \
+        'transformers>=4.37.0' \
         accelerate \
         qwen-vl-utils[decord]==0.0.8 \
         fastapi \
         uvicorn[standard] \
         python-multipart \
         pillow \
-        pydantic
+        pydantic \
+        supervision && \
+    # Install AutoAWQ last
+    pip install --no-cache-dir autoawq
 
 # Copy application files
 COPY --chown=user:user . /app
 
-# Create cache directories and set permissions
+# Create required directories and set permissions
 RUN mkdir -p /home/user/.cache/huggingface && \
     mkdir -p /temp && \
     chown -R user:user /home/user/.cache && \
@@ -49,7 +50,8 @@ USER user
 ENV HOME=/home/user \
     PATH=/home/user/.local/bin:$PATH \
     TRANSFORMERS_CACHE=/home/user/.cache/huggingface \
-    TORCH_HOME=/home/user/.cache/torch
+    TORCH_HOME=/home/user/.cache/torch \
+    HF_HOME=/home/user/.cache/huggingface
 
 # Expose the port
 EXPOSE 7860