Update Dockerfile

Dockerfile

@@ -13,14 +13,15 @@ RUN apt-get update && apt-get install -y \
 RUN useradd -m -u 1000 user
 WORKDIR /app
 
+
 # Install Python dependencies in correct order
 RUN pip install --no-cache-dir --upgrade pip && \
     # 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 \
+        torch \
+        torchvision  \
         ninja \
         transformers \
         accelerate \