Update Dockerfile

Dockerfile

@@ -14,13 +14,22 @@ RUN pip install --no-cache-dir -r requirements.txt
 # Copy application code
 COPY . .
 
-# Set environment variables for cache dirs (Docker-friendly)
-ENV HF_HOME=/tmp
-ENV TRANSFORMERS_CACHE=/tmp
-ENV TORCHINDUCTOR_CACHE_DIR=/tmp/torch_inductor_cache
+# # Set environment variables for cache dirs (Docker-friendly)
+# ENV HF_HOME=/tmp
+# ENV TRANSFORMERS_CACHE=/tmp
+# ENV TORCHINDUCTOR_CACHE_DIR=/tmp/torch_inductor_cache
+
+# # Create cache folder
+# RUN mkdir -p /tmp/torch_inductor_cache
+# Add a user
+RUN useradd -m -u 1000 appuser
+USER appuser
+WORKDIR /app
 
-# Create cache folder
-RUN mkdir -p /tmp/torch_inductor_cache
+# Set env variables for torch caches
+ENV TORCH_HOME=/tmp/torch_home
+ENV TORCHINDUCTOR_CACHE_DIR=/tmp/torch_inductor_cache
+RUN mkdir -p /tmp/torch_home /tmp/torch_inductor_cache
 
 # Expose port
 EXPOSE 7860