Upload 4 files

Dockerfile

@@ -2,11 +2,15 @@ FROM python:3.11-slim
 
 WORKDIR /app
 
-# Install system dependencies
+# Install system dependencies including Ollama
 RUN apt-get update && apt-get install -y \
     curl \
+    wget \
     && rm -rf /var/lib/apt/lists/*
 
+# Install Ollama
+RUN curl -fsSL https://ollama.ai/install.sh | sh
+
 # Copy requirements and install Python dependencies
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
@@ -14,6 +18,10 @@ RUN pip install --no-cache-dir -r requirements.txt
 # Copy application code
 COPY . .
 
+# Create a startup script
+RUN echo '#!/bin/bash\nollama serve &\nsleep 5\npython3 -m gunicorn --bind 0.0.0.0:7860 --workers 1 --timeout 120 app:app' > /app/start.sh && \
+    chmod +x /app/start.sh
+
 # Expose port
 EXPOSE 7860
 
@@ -21,5 +29,5 @@ EXPOSE 7860
 HEALTHCHECK --interval=30s --timeout=10s --start-period=5s --retries=3 \
     CMD curl -f http://localhost:7860/health || exit 1
 
-# Run the application
-CMD ["gunicorn", "--bind", "0.0.0.0:7860", "--workers", "1", "--timeout", "120", "app:app"]
+# Run the startup script
+CMD ["/app/start.sh"]

README.md

@@ -25,9 +25,8 @@ A Hugging Face Space that provides a REST API interface for Ollama models, allow
 
 1. Fork this repository or create a new Space
 2. Upload these files to your Space
-3. Set the following environment variables in your Space settings:
-   - `OLLAMA_BASE_URL`: URL to your Ollama instance (e.g., `http://localhost:11434`)
-   - `ALLOWED_MODELS`: Comma-separated list of allowed models (optional)
+3. **No environment variables needed** - Ollama runs inside the Space!
+4. Wait for the build to complete (may take 10-15 minutes due to Ollama installation)
 
 ### 2. Local Development
 
@@ -39,8 +38,11 @@ cd ollama-space
 # Install dependencies
 pip install -r requirements.txt
 
-# Set environment variables
-export OLLAMA_BASE_URL=http://localhost:11434
+# Install Ollama locally
+curl -fsSL https://ollama.ai/install.sh | sh
+
+# Start Ollama in another terminal
+ollama serve
 
 # Run the application
 python app.py
@@ -121,10 +123,12 @@ Health check endpoint.
 
 ### Environment Variables
 
-- `OLLAMA_BASE_URL`: URL to your Ollama instance (default: `http://localhost:11434`)
+- `OLLAMA_BASE_URL`: URL to your Ollama instance (default: `http://localhost:11434` - **Ollama runs inside this Space!**)
 - `MODELS_DIR`: Directory for storing models (default: `/models`)
 - `ALLOWED_MODELS`: Comma-separated list of allowed models (default: all models)
 
+**Note**: This Space now includes Ollama installed directly inside it, so you don't need an external Ollama instance!
+
 ### Supported Models
 
 By default, the following models are allowed: