Update Dockerfile

Dockerfile

@@ -1,21 +1,22 @@
-FROM python:3.9-slim
+# Use the official lightweight Python image
+FROM python:3.10-slim
 
+# Set working directory
 WORKDIR /app
 
-RUN apt-get update && apt-get install -y \
-    build-essential \
-    curl \
-    software-properties-common \
-    git \
-    && rm -rf /var/lib/apt/lists/*
+# Copy source files
+COPY ./src /app/src
+COPY requirements.txt .
 
-COPY requirements.txt ./
-COPY src/ ./src/
+# Create a writable cache directory for transformers
+RUN mkdir -p /app/cache
 
-RUN pip3 install -r requirements.txt
+# Install dependencies
+RUN pip install --no-cache-dir -r requirements.txt
 
-EXPOSE 8501
+# Set environment variables
+ENV TRANSFORMERS_CACHE=/app/cache
+ENV STREAMLIT_SERVER_HEADLESS=true
 
-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"]
+# Run the app using Streamlit
+CMD ["streamlit", "run", "src/streamlit_app.py", "--server.port=7860", "--server.address=0.0.0.0"]