Create Dockerfile

Dockerfile

@@ -0,0 +1,31 @@
+# Use a lightweight Python base
+FROM python:3.10-slim
+
+# Set working directory
+WORKDIR /app
+
+# Install system tools (curl is needed to install Ollama)
+RUN apt-get update && apt-get install -y curl && rm -rf /var/lib/apt/lists/*
+
+# Install Ollama
+RUN curl -fsSL https://ollama.com/install.sh | sh
+
+# Install Python API requirements
+RUN pip install --no-cache-dir fastapi uvicorn requests
+
+# Copy your API code
+COPY . .
+
+# Create a start script to handle the background process
+RUN echo '#!/bin/bash' > start.sh && \
+    echo 'ollama serve &' >> start.sh && \
+    echo 'sleep 5' >> start.sh && \
+    echo 'ollama pull granite-code:2b' >> start.sh && \
+    echo 'uvicorn api:app --host 0.0.0.0 --port 7860' >> start.sh && \
+    chmod +x start.sh
+
+# Hugging Face expects port 7860
+EXPOSE 7860
+
+# Run the start script
+CMD ["./start.sh"]