Update Dockerfile

Dockerfile

@@ -4,6 +4,7 @@ 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
 
 # Create the cache directory and set permissions
 RUN mkdir -p /tmp/.cache && chmod -R 777 /tmp/.cache