Update Dockerfile

Dockerfile

@@ -1,7 +1,10 @@
-FROM python:3.9-slim
+# Use official Python slim image
+FROM python:3.10-slim
 
+# Set working directory
 WORKDIR /app
 
+# Install system dependencies (for transformers and torch)
 RUN apt-get update && apt-get install -y \
     build-essential \
     curl \
@@ -9,13 +12,23 @@ RUN apt-get update && apt-get install -y \
     git \
     && rm -rf /var/lib/apt/lists/*
 
+# Copy requirements and source code
 COPY requirements.txt ./
 COPY src/ ./src/
 
-RUN pip3 install -r requirements.txt
+# Upgrade pip and install dependencies
+RUN pip install --upgrade pip
+RUN pip install -r requirements.txt
 
+# Expose ports for Streamlit (8501) and FastAPI (8000)
 EXPOSE 8501
+EXPOSE 8000
 
-HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health
+# Set environment variable for Huggingface cache
+ENV HF_HOME=/tmp/huggingface_cache
 
-ENTRYPOINT ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0"]
+# Healthcheck for Streamlit
+HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health || exit 1
+
+# Run the Streamlit app (which starts FastAPI in background)
+ENTRYPOINT ["streamlit", "run", "src/streamlit.py", "--server.port=8501", "--server.address=0.0.0.0"]