Update Dockerfile

Dockerfile

@@ -2,24 +2,29 @@ FROM python:3.9-slim
 
 WORKDIR /app
 
-# Install dependencies
+# Install system dependencies
 RUN apt-get update && apt-get install -y \
     build-essential \
-    curl \
-    software-properties-common \
-    git \
+    libgl1 \
     && rm -rf /var/lib/apt/lists/*
 
-# Create a writable cache directory and give permissions
-RUN mkdir -p /app/.cache && chmod -R 777 /app/.cache
+# Create cache directory
+RUN mkdir -p /cache && chmod -R 777 /cache
 
-COPY requirements.txt ./
-COPY src/ ./src/
+# Copy requirements first for better caching
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
 
-RUN pip3 install -r requirements.txt
+# Copy application
+COPY src/ ./src/
 
+# Set environment variables
+ENV TRANSFORMERS_CACHE=/cache
+ENV STREAMLIT_SERVER_PORT=8501
+ENV STREAMLIT_SERVER_MAX_UPLOAD_SIZE=200
 EXPOSE 8501
 
-HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health
-
-ENTRYPOINT ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0", "--server.maxUploadSize=200"]
+CMD ["streamlit", "run", "src/streamlit_app.py", \
+     "--server.port=8501", \
+     "--server.address=0.0.0.0", \
+     "--server.maxUploadSize=200"]