Use PyTorch base image - faster build

Dockerfile

@@ -1,10 +1,11 @@
 # =============================================================================
-# FarmEyes - HuggingFace Spaces Dockerfile (GPU - Pre-built Wheel)
+# FarmEyes - HuggingFace Spaces Dockerfile (GPU - Simplified)
 # =============================================================================
-# Uses pre-compiled llama-cpp-python to avoid build timeout
+# Uses PyTorch base image with CUDA pre-installed
+# llama-cpp-python CPU version works fine - still faster than free tier
 # =============================================================================
 
-FROM nvidia/cuda:11.8.0-runtime-ubuntu22.04
+FROM pytorch/pytorch:2.1.0-cuda11.8-cudnn8-runtime
 
 # Set environment variables
 ENV PYTHONUNBUFFERED=1
@@ -14,36 +15,23 @@ ENV DEBIAN_FRONTEND=noninteractive
 
 WORKDIR /app
 
-# Install Python and system dependencies
+# Install system dependencies only
 RUN apt-get update && apt-get install -y --no-install-recommends \
-    python3.10 \
-    python3-pip \
-    python3.10-dev \
     ffmpeg \
     libsm6 \
     libxext6 \
     libgl1 \
-    curl \
     && rm -rf /var/lib/apt/lists/*
 
-# Make python3.10 the default
-RUN ln -sf /usr/bin/python3.10 /usr/bin/python && \
-    ln -sf /usr/bin/pip3 /usr/bin/pip
-
-# Upgrade pip
-RUN pip install --no-cache-dir --upgrade pip setuptools wheel
-
-# Install PyTorch with CUDA 11.8
-RUN pip install --no-cache-dir torch torchvision --index-url https://download.pytorch.org/whl/cu118
-
-# Install llama-cpp-python PRE-BUILT wheel (no compilation!)
-RUN pip install --no-cache-dir llama-cpp-python \
-    --extra-index-url https://abetlen.github.io/llama-cpp-python/whl/cu118
-
-# Copy and install remaining requirements
+# Copy requirements
 COPY requirements.txt .
+
+# Install Python dependencies (torch already in base image)
 RUN pip install --no-cache-dir -r requirements.txt
 
+# Install llama-cpp-python (CPU version - avoids long compile)
+RUN pip install --no-cache-dir llama-cpp-python
+
 # Copy application code
 COPY . .
 

requirements.txt

@@ -1,18 +1,19 @@
+# =============================================================================
+# FarmEyes - Requirements (GPU Version)
+# =============================================================================
+# Note: torch/torchvision already in base Docker image
+# =============================================================================
+
 # Web Framework
 fastapi>=0.104.0
 uvicorn[standard]>=0.24.0
 python-multipart>=0.0.6
 
-# AI/ML
-torch>=2.0.0
-torchvision>=0.15.0
+# AI/ML (torch already in base image)
 ultralytics>=8.0.0
 transformers>=4.35.0
 huggingface-hub>=0.19.0
 
-# GGUF Model Support (for N-ATLaS)
-llama-cpp-python>=0.2.0
-
 # Audio
 openai-whisper>=20231117
 soundfile>=0.12.0
@@ -28,4 +29,4 @@ requests>=2.31.0
 numpy>=1.24.0
 scipy>=1.11.0
 pydantic>=2.0.0
-python-dotenv>=1.0.0
+python-dotenv>=1.0.0