Create Dockerfile

Dockerfile

@@ -0,0 +1,15 @@
+# Use the official Hugging Face Spaces GPU image
+FROM huggingface/zero-gpu-gradio:latest
+
+# Build llama-cpp-python with CUDA support for the RTX Pro 6000 Blackwell
+ENV CMAKE_ARGS="-DGGML_CUDA=on"
+ENV FORCE_CMAKE=1
+
+# Install the Python dependencies (this triggers the llama-cpp-python CUDA build)
+COPY requirements.txt /tmp/requirements.txt
+RUN pip install --no-cache-dir -r /tmp/requirements.txt
+
+# Copy the app
+COPY app.py /home/user/app/app.py
+
+CMD ["python", "app.py"]