Update Dockerfile

Dockerfile

@@ -1,28 +1,36 @@
-FROM python:3.11-slim
+FROM python:3.10-slim
 
-RUN apt-get update && apt-get install -y --no-install-recommends \
-        libgl1 libglib2.0-0 libsm6 libxext6 libxrender-dev curl \
+RUN apt-get update && apt-get install -y \
+    build-essential \
+    curl \
+    libgl1 \
+    libglib2.0-0 \
     && rm -rf /var/lib/apt/lists/*
 
-WORKDIR /app
+RUN useradd -m -u 1000 appuser
 
-# CPU PyTorch
-RUN pip install --no-cache-dir \
-    torch==2.2.2+cpu torchvision==0.17.2+cpu \
-    --extra-index-url https://download.pytorch.org/whl/cpu
+WORKDIR /app
 
 COPY requirements.txt .
-RUN pip install --no-cache-dir -r requirements.txt
-
-COPY app.py .
+RUN pip install --no-cache-dir --upgrade "pip==24.3.1" && \
+    pip install --no-cache-dir -r requirements.txt
 
+COPY . .
 
+RUN chown -R appuser:appuser /app
 USER appuser
+
+ENV STREAMLIT_SERVER_PORT=7860
+ENV STREAMLIT_SERVER_ADDRESS=0.0.0.0
+ENV STREAMLIT_SERVER_HEADLESS=true
+ENV STREAMLIT_BROWSER_GATHER_USAGE_STATS=false
+ENV HF_HOME=/app/.cache/huggingface
+ENV TRANSFORMERS_CACHE=/app/.cache/huggingface
+
 EXPOSE 7860
 
 CMD ["streamlit", "run", "app.py", \
      "--server.port=7860", \
      "--server.address=0.0.0.0", \
      "--server.headless=true", \
-     "--server.fileWatcherType=none", \
-     "--browser.gatherUsageStats=false"]
+     "--browser.gatherUsageStats=false"]