:lady_beetle: Fix cache folder permissions

Dockerfile

@@ -1,22 +1,36 @@
 # Use the official PyTorch Docker image as a base (includes CUDA and PyTorch)
 FROM pytorch/pytorch:1.11.0-cuda11.3-cudnn8-runtime
 
-# Set a working directory in the container
-WORKDIR /workspace
+# Install required dependencies (add any additional system dependencies you need)
+RUN apt update && apt install -y ffmpeg
 
-# Create a writable cache directory
-RUN mkdir -p /workspace/cache
+# Create a non-root user with a home directory
+RUN useradd -m -u 1000 user
 
-# Set the environment variable for cache
-ENV TRANSFORMERS_CACHE=/workspace/cache
+# Switch to the new non-root user
+USER user
 
-# Install Python dependencies
+# Set environment variables for the new user
+ENV HOME=/home/user \
+    PATH=/home/user/.local/bin:$PATH
+
+# Set a working directory
+WORKDIR $HOME/app
+
+# Set the TRANSFORMERS_CACHE directory to be within the user's home directory
+ENV TRANSFORMERS_CACHE=$HOME/cache
+
+# Copy the app code and set ownership to the non-root user
+COPY --chown=user . $HOME/app
+
+# Install Python dependencies in the virtual environment
+RUN python -m venv /home/user/venv
+ENV PATH="/home/user/venv/bin:$PATH"
+
+# Install pip dependencies within the virtual environment
 COPY requirements.txt .
 RUN pip install --upgrade pip
 RUN pip install -r requirements.txt
 
-# Copy the training script
-COPY app.py .
-
 # Run the training script
 CMD ["python", "app.py"]