Update Dockerfile

Dockerfile

@@ -4,7 +4,9 @@ FROM python:3.9-slim
 # Set environment variables for Hugging Face cache
 ENV TRANSFORMERS_CACHE=/tmp/.cache
 ENV HF_DATASETS_CACHE=/tmp/.cache
-ENV TORCH_HOME=/tmp/.cache  # Set PyTorch cache directory
+
+# Set PyTorch cache directory
+ENV TORCH_HOME=/tmp/.cache
 
 # Create the cache directory and set permissions
 RUN mkdir -p /tmp/.cache && chmod -R 777 /tmp/.cache