Update Dockerfile

Dockerfile

@@ -1,33 +1,38 @@
-# ---- Base image
+# Use a lightweight Python base image (CPU only)
 FROM python:3.11-slim
 
-# ---- System deps
-RUN apt-get update && apt-get install -y --no-install-recommends \
-    git build-essential && \
-    rm -rf /var/lib/apt/lists/*
-
-# ---- Python deps
-# By default this installs CPU Torch. If you build on a GPU runner and want CUDA,
-# uncomment the second pip line and comment out the first.
-RUN pip install --no-cache-dir \
-    transformers accelerate torch sentencepiece safetensors gradio==4.44.0
-
-# For CUDA builds (optional):
-# RUN pip install --no-cache-dir \
-#     transformers accelerate sentencepiece safetensors gradio==4.44.0 && \
-#     pip install --no-cache-dir --index-url https://download.pytorch.org/whl/cu121 torch==2.3.1+cu121
-
-# ---- Cache & runtime env
-ENV HF_HOME=/root/.cache/huggingface \
+# Environment variables
+ENV HF_HOME=/app/hf_cache \
+    TRANSFORMERS_CACHE=/app/hf_cache \
     GRADIO_SERVER_NAME=0.0.0.0 \
     GRADIO_SERVER_PORT=7860 \
     PYTHONUNBUFFERED=1 \
     TOKENIZERS_PARALLELISM=false
 
-# ---- App files
+# Set working directory
 WORKDIR /app
-COPY app.py chat.py /app/
 
+# Install system dependencies (needed by torch/transformers)
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    git \
+    wget \
+    build-essential \
+    && rm -rf /var/lib/apt/lists/*
+
+# Create cache directory with open permissions
+RUN mkdir -p /app/hf_cache && chmod -R 777 /app/hf_cache
+
+# Copy requirements first for Docker cache efficiency
+COPY requirements.txt .
+
+# Install Python dependencies
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Copy the rest of the app
+COPY . .
+
+# Expose Gradio port
 EXPOSE 7860
 
+# Run the app
 CMD ["python", "app.py"]