Update Dockerfile

Dockerfile

@@ -1,8 +1,11 @@
-# Use official Python image as base
+# Use lightweight Python image
 FROM python:3.9-slim
 
-# Set working directory
+# Set up environment
 WORKDIR /app
+ENV PYTHONUNBUFFERED=1 \
+    TRANSFORMERS_CACHE=/app/model_cache \
+    HF_HOME=/app/model_cache
 
 # Install system dependencies
 RUN apt-get update && \
@@ -11,27 +14,20 @@ RUN apt-get update && \
     python3-dev \
     && rm -rf /var/lib/apt/lists/*
 
-# Create cache directory with proper permissions
-RUN mkdir -p /app/model_cache && chmod a+rwx /app/model_cache
+# Create cache directory with write permissions
+RUN mkdir -p /app/model_cache && chmod 777 /app/model_cache
 
-# Set environment variables for cache locations
-ENV TRANSFORMERS_CACHE=/app/model_cache
-ENV HF_HOME=/app/model_cache
-
-# Copy requirements first to leverage Docker cache
+# Copy only necessary files
 COPY requirements.txt .
-
-# Install Python dependencies
-RUN pip install --no-cache-dir -r requirements.txt
-
-# Copy the application
 COPY app.py .
 
-# Expose the port Streamlit runs on
-EXPOSE 8501
+# Install Python packages
+RUN pip install --no-cache-dir -r requirements.txt && \
+    python -c "from transformers import pipeline; pipeline('question-answering', model='distilbert-base-uncased-distilled-squad')"
 
-# Health check
-HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health || exit 1
+# Expose and run
+EXPOSE 8501
+HEALTHCHECK --interval=30s --timeout=30s --start-period=5s --retries=3 \
+    CMD curl -f http://localhost:8501/_stcore/health || exit 1
 
-# Run the application
 CMD ["streamlit", "run", "app.py", "--server.port=8501", "--server.address=0.0.0.0"]