Upgrade to GPU with GGUF model support

Dockerfile

@@ -1,32 +1,48 @@
 # =============================================================================
-# FarmEyes - HuggingFace Spaces Dockerfile
+# FarmEyes - HuggingFace Spaces Dockerfile (GPU Version)
 # =============================================================================
 
-FROM python:3.10-slim
-
-WORKDIR /app
+FROM nvidia/cuda:11.8.0-devel-ubuntu22.04
 
+# Set environment variables
 ENV PYTHONUNBUFFERED=1
 ENV HOST=0.0.0.0
 ENV PORT=7860
-ENV PIP_NO_CACHE_DIR=1
+ENV DEBIAN_FRONTEND=noninteractive
+
+WORKDIR /app
 
-# Install system dependencies
+# Install Python and system dependencies
 RUN apt-get update && apt-get install -y --no-install-recommends \
+    python3.10 \
+    python3-pip \
+    python3.10-dev \
     ffmpeg \
     libsm6 \
     libxext6 \
     libgl1 \
+    build-essential \
+    cmake \
+    git \
     && rm -rf /var/lib/apt/lists/*
 
-# Copy and install requirements
+# Make python3.10 the default
+RUN ln -sf /usr/bin/python3.10 /usr/bin/python
+
+# Copy requirements
 COPY requirements.txt .
-RUN pip install --upgrade pip && \
-    pip install -r requirements.txt
+
+# Upgrade pip and install dependencies
+RUN pip install --no-cache-dir --upgrade pip && \
+    pip install --no-cache-dir -r requirements.txt
+
+# Install llama-cpp-python with CUDA support
+RUN CMAKE_ARGS="-DLLAMA_CUBLAS=on" pip install --no-cache-dir llama-cpp-python
 
 # Copy application code
 COPY . .
 
+# Create directories
 RUN mkdir -p /app/uploads /app/temp
 
 EXPOSE 7860

requirements.txt

@@ -10,6 +10,9 @@ 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