Add Ollama Docker Space with gemma3:2b

Dockerfile

@@ -1,30 +1,12 @@
-FROM golang:1.21 AS builder
+FROM ollama/ollama:latest
 
-WORKDIR /build
-COPY go.mod .
-COPY main.go .
-RUN go build -o ollama-proxy .
+RUN apt-get update && apt-get install -y curl && rm -rf /var/lib/apt/lists/*
 
-FROM ubuntu:22.04
-
-ENV DEBIAN_FRONTEND=noninteractive
-ENV OLLAMA_HOST=0.0.0.0
-
-RUN apt-get update && apt-get install -y \
-    curl \
-    ca-certificates \
-    zstd \
-    && rm -rf /var/lib/apt/lists/*
-
-RUN curl -fsSL https://ollama.com/install.sh | sh
-
-WORKDIR /app
-COPY --from=builder /build/ollama-proxy .
-COPY start.sh .
-RUN chmod +x start.sh
+COPY start.sh /start.sh
+RUN chmod +x /start.sh
 
 EXPOSE 7860
 
-ENV PORT=7860
+ENV OLLAMA_HOST=0.0.0.0:7860
 
-CMD ["./start.sh"]
+ENTRYPOINT ["/start.sh"]

README.md

@@ -1,10 +1,12 @@
 ---
 title: Ollama Proxy
-emoji: 🐨
+emoji: 🤖
 colorFrom: blue
-colorTo: yellow
+colorTo: purple
 sdk: docker
 pinned: false
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+# Ollama Proxy Space
+
+Runs Ollama with gemma3:2b model via Docker.

start.sh

@@ -1,22 +1,16 @@
 #!/bin/bash
 set -e
 
-echo "Starting Ollama service..."
 ollama serve &
-OLLAMA_PID=$!
+SERVER_PID=$!
 
-echo "Waiting for Ollama to be ready..."
-for i in $(seq 1 30); do
-    if curl -sf http://localhost:11434/api/tags > /dev/null 2>&1; then
-        echo "Ollama is ready"
-        break
-    fi
-    echo "Waiting... ($i/30)"
-    sleep 2
+echo "Waiting for Ollama to start..."
+until curl -s http://localhost:7860/api/tags > /dev/null 2>&1; do
+  sleep 2
 done
 
-echo "Pulling model ingu627/exaone4.0:1.2b ..."
-ollama pull ingu627/exaone4.0:1.2b
+echo "Pulling gemma3:2b model..."
+ollama pull gemma3:2b
 
-echo "Starting proxy on port ${PORT:-7860}..."
-exec ./ollama-proxy
+echo "Ollama ready with gemma3:2b"
+wait $SERVER_PID