Update Dockerfile

Dockerfile

@@ -1,8 +1,10 @@
+# Use Python 3.9 slim as the base image
 FROM python:3.9-slim
 
+# Set working directory
 WORKDIR /app
 
-# Install necessary dependencies for OpenCV and other utilities
+# Install necessary dependencies for OpenCV, PyTorch, and other utilities
 RUN apt-get update && apt-get install -y \
     build-essential \
     curl \
@@ -12,19 +14,27 @@ RUN apt-get update && apt-get install -y \
     libglib2.0-0 \
     && rm -rf /var/lib/apt/lists/*
 
-# Set environment variable for Streamlit config
-ENV STREAMLIT_CONFIG=/app/.streamlit
+# Create a non-root user to avoid permission issues
+RUN useradd -m -u 1000 appuser && chown -R appuser:appuser /app
+USER appuser
 
-# Create the .streamlit directory
-RUN mkdir -p /app/.streamlit
+# Set environment variables for Streamlit and Hugging Face
+ENV STREAMLIT_CONFIG_DIR=/app/.streamlit
+ENV TRANSFORMERS_CACHE=/app/.cache/huggingface
+ENV HF_HOME=/app/.cache/huggingface
+ENV STREAMLIT_SERVER_FILE_WATCHER_TYPE=none
+
+# Create necessary directories with correct permissions
+RUN mkdir -p /app/.streamlit /app/.cache/huggingface
+RUN chmod -R 777 /app/.streamlit /app/.cache
 
 # Copy the requirements and install them
-COPY requirements.txt ./
-RUN pip3 install -r requirements.txt
+COPY --chown=appuser:appuser requirements.txt ./
+RUN pip3 install --no-cache-dir -r requirements.txt
 
 # Copy the source code and modules directory
-COPY src/ ./src/
-COPY modules/ ./modules/
+COPY --chown=appuser:appuser src/ ./src/
+COPY --chown=appuser:appuser modules/ ./modules/
 
 # Expose the port that Streamlit will run on
 EXPOSE 8501
@@ -33,4 +43,4 @@ EXPOSE 8501
 HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health || exit 1
 
 # Set the entry point to run the Streamlit app
-ENTRYPOINT ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0"]
+ENTRYPOINT ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0"]