update backend for HF

Dockerfile

@@ -1,32 +1,38 @@
-# 1. Use Python 3.10
+# 1. Use Python 3.10 slim
 FROM python:3.10-slim
 
 # 2. Set working directory
 WORKDIR /app
 
-# 3. Install system dependencies (including build tools for llama-cpp-python if needed)
-RUN apt-get update && apt-get install -y \
+# 3. Install minimal system dependencies
+#    - libgl1/libglib2.0-0: OpenCV needs these
+#    - build-essential/python3-dev: llama-cpp-python builds from source
+RUN apt-get update && apt-get install -y --no-install-recommends \
     libgl1 \
     libglib2.0-0 \
     build-essential \
     python3-dev \
     && rm -rf /var/lib/apt/lists/*
 
-# 4. Install Python dependencies
+# 4. Install PyTorch CPU-only FIRST (saves ~1.5GB vs full CUDA torch)
+#    This is a separate layer so it caches well
+RUN pip install --no-cache-dir \
+    torch --index-url https://download.pytorch.org/whl/cpu
+
+# 5. Install remaining Python dependencies
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
-# 5. Copy your code
+# 6. Copy your code
 COPY . .
 
-# 6. Download all ML models (Face, Text, LLaMA 3 GGUF) during build
-# This ensures a "batteries included" image for HF Spaces
+# 7. Download all ML models (Face, Text, LLaMA 3 GGUF) during build
 RUN python download_models.py
 
-# 7. Environment & Port settings (7860 is HF Spaces standard)
+# 8. Environment & Port settings (7860 is HF Spaces standard)
 ENV PYTHONPATH=/app
 ENV USE_EMBEDDED_LLM=True
 EXPOSE 7860
 
-# 8. Run the app with Uvicorn
+# 9. Run the app with Uvicorn
 CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "7860"]

requirements.txt

@@ -12,18 +12,16 @@ anyio>=4.0.0
 # --- Rate Limiting ---
 slowapi>=0.1.9
 
-# --- AI & Vision (Preserved - Version Locked for Stability) ---
+# --- AI & Vision (CPU-only TensorFlow — saves ~500MB vs full tensorflow) ---
 numpy<2.0
-opencv-python
-tensorflow
+opencv-python-headless
 tensorflow-cpu
 pandas
 pillow
 gdown
 
-# --- NLP (New) ---
+# --- NLP (PyTorch CPU-only via --index-url, see Dockerfile) ---
 transformers>=4.40.0
-torch>=2.0.0
 sentencepiece==0.1.99
 llama-cpp-python>=0.2.77
 huggingface-hub>=0.23.0