Update Dockerfile

Dockerfile

@@ -12,7 +12,9 @@ RUN pip install --no-cache-dir \
     datasets \
     tenacity \
     safetensors
-# For retry logic
+
+# Avoids installing GPU-specific builds that won’t work in CPU-only environments.
+RUN pip install --no-cache-dir torch==2.1.2+cpu -f https://download.pytorch.org/whl/torch_stable.html
 
 # Configure cache
 ENV TRANSFORMERS_CACHE="/cache/models"