Upload Dockerfile

Dockerfile

@@ -1,11 +1,14 @@
 FROM python:3.11-slim
 
 WORKDIR /app
-# Install pre-built wheels to avoid C++ compilation memory limit (OOMKilled)
-RUN apt-get update && apt-get install -y curl && rm -rf /var/lib/apt/lists/*
+# Install build tools for llama-cpp-python (C++ compiler)
+RUN apt-get update && apt-get install -y g++ build-essential cmake curl && rm -rf /var/lib/apt/lists/*
 
 COPY requirements.txt .
-RUN pip install --no-cache-dir -r requirements.txt --extra-index-url https://abetlen.github.io/llama-cpp-python/whl/cpu
+# Force single-threaded compilation to prevent OOMKilled on Hugging Face Spaces
+ENV CMAKE_BUILD_PARALLEL_LEVEL=1
+ENV MAKEFLAGS="-j1"
+RUN pip install --no-cache-dir -r requirements.txt
 COPY . .
 
 # Hugging Face exposes port 7860