Update Dockerfile

Dockerfile

@@ -1,20 +1,24 @@
 
-FROM huggingface/transformers-pytorch-gpu:latest
+FROM python:3.9-slim  # Lightweight CPU-only base
 
-# Install with backoff-retry for robustness
-RUN pip install --no-cache-dir \
-    gradio \
-    torch \
-    transformers \
-    sentencepiece \
-    accelerate \
-    huggingface-hub \
-    datasets \
-    tenacity \
-    safetensors
+# Install system dependencies
+RUN apt-get update && apt-get install -y \
+    git \
+    gcc \
+    g++ \
+    && rm -rf /var/lib/apt/lists/*
 
-# Avoids installing GPU-specific builds that won’t work in CPU-only environments.
-RUN pip install --no-cache-dir torch==2.1.2+cpu -f https://download.pytorch.org/whl/torch_stable.html
+# Install Python packages (CPU-only versions)
+RUN pip install --no-cache-dir \
+    torch==2.1.2+cpu -f https://download.pytorch.org/whl/torch_stable.html \
+    transformers==4.36.0 \
+    sentencepiece==0.2.0 \
+    accelerate==0.25.0 \
+    huggingface-hub==0.19.4 \
+    gradio==3.50.2 \
+    datasets==2.15.0 \
+    safetensors==0.4.1 \
+    tenacity==8.2.3  # Explicitly added with version
 
 # Configure cache
 ENV TRANSFORMERS_CACHE="/cache/models"
@@ -24,7 +28,4 @@ RUN mkdir -p /cache/models /cache/datasets
 WORKDIR /app
 COPY . /app
 
-# Pre-cache during build (optional)
-# RUN python cache_to_repo.py
-
 CMD ["python", "app.py"]