Update Dockerfile

Dockerfile

@@ -1,22 +1,26 @@
-# Use the official PyTorch base image with a confirmed tag (CPU-only, Python 3.10)
-FROM pytorch/pytorch:2.0.1-cpu-py310
+# Go back to a stable, general Python image
+FROM python:3.10
 
 WORKDIR /app
 
-# Install necessary system packages for Streamlit and Git
+# Install necessary system packages (build essentials are critical for sentencepiece)
 RUN apt-get update && apt-get install -y \
     curl \
     git \
+    build-essential \
     libffi-dev \
     libssl-dev \
+    # Clean up to save space
     && rm -rf /var/lib/apt/lists/*
 
-# Copy and install ONLY non-torch dependencies (torch is already installed)
+# Install the problematic package (transformers/sentencepiece) using the special flag
 COPY requirements.txt ./
-# Install non-torch dependencies, ignoring any torch/cuda mentions
-RUN pip3 install --no-cache-dir --force-reinstall -r requirements.txt
+# Install non-torch dependencies, then install transformers with sentencepiece extra
+RUN pip install --no-cache-dir --upgrade pip && \
+    pip install --no-cache-dir "transformers[sentencepiece]" && \
+    pip install --no-cache-dir -r requirements.txt
 
-# Copy application source code
+# Copy application source code (your streamlit_app.py)
 COPY src/ ./src/
 
 EXPOSE 8501