Fix: set HOME/USER/TORCH env vars for uid 1000

Dockerfile

@@ -2,8 +2,17 @@ FROM python:3.10-slim
 
 WORKDIR /app
 
+# HF Spaces runs as uid 1000 - set all cache dirs to writable locations
 ENV HF_HOME=/tmp/hf_cache
 ENV TRANSFORMERS_CACHE=/tmp/hf_cache
+ENV TORCH_HOME=/tmp/torch_cache
+ENV TORCHINDUCTOR_CACHE_DIR=/tmp/torch_cache
+ENV XDG_CACHE_HOME=/tmp/cache
+ENV HOME=/tmp/home
+ENV USER=user
+
+RUN mkdir -p /tmp/hf_cache /tmp/torch_cache /tmp/cache /tmp/home && \
+    chmod -R 777 /tmp/hf_cache /tmp/torch_cache /tmp/cache /tmp/home
 
 RUN pip install --no-cache-dir \
     torch \
@@ -15,7 +24,6 @@ RUN pip install --no-cache-dir \
     huggingface_hub
 
 COPY train_on_hf.py .
-
 RUN chmod -R 777 /app
 
 CMD ["sh", "-c", "python train_on_hf.py --hf-token $HF_TOKEN"]