fix bug

Dockerfile

@@ -1,10 +1,32 @@
 FROM python:3.9
 
+# Create a non-root user
+RUN useradd -m -u 1000 user
+
+# Create cache directory and set permissions
+RUN mkdir -p /home/user/.cache/huggingface && \
+    mkdir -p /home/user/.cache/torch && \
+    chown -R user:user /home/user/.cache
+
+# Set environment variables
+ENV TRANSFORMERS_CACHE=/home/user/.cache/huggingface
+ENV TORCH_HOME=/home/user/.cache/torch
+ENV HF_HOME=/home/user/.cache/huggingface
+
 WORKDIR /code
 
+# Copy and install requirements
 COPY ./requirements.txt /code/requirements.txt
 RUN pip install --no-cache-dir -r requirements.txt
 
+# Copy application code
 COPY ./app.py /code/app.py
 
+# Change ownership of the application directory
+RUN chown -R user:user /code
+
+# Switch to non-root user
+USER user
+
+# Run the application
 CMD ["python", "app.py"]