Update Dockerfile

Dockerfile

@@ -2,7 +2,6 @@ FROM python:3.9-slim
 
 WORKDIR /app
 
-# Install dependencies
 RUN apt-get update && apt-get install -y \
     build-essential \
     curl \
@@ -14,15 +13,18 @@ RUN apt-get update && apt-get install -y \
 COPY requirements.txt ./
 COPY src/ ./src/
 
-# Upgrade pip and install dependencies
+# Upgrade pip and install dependencies (without transformers)
 RUN pip install --upgrade pip && \
     pip install --no-cache-dir -r requirements.txt
 
-# Confirm transformers and torch version
-RUN python -c "import transformers; import torch; print('Transformers:', transformers.__version__, 'Torch:', torch.__version__)"
+# Force reinstall transformers from PyPI (latest stable)
+RUN pip install --no-cache-dir --upgrade "transformers>=4.40.1"
+
+# Optional: print installed version (debugging)
+RUN python -c "import transformers; print('Transformers version:', transformers.__version__)"
 
 EXPOSE 8501
 
 HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health || exit 1
 
-ENTRYPOINT ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0"]
+ENTRYPOINT ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0"]