Update Dockerfile

Dockerfile

@@ -1,15 +1,24 @@
-# Use the official Ollama image as the base
-FROM ollama/ollama:latest
+# Start with a lightweight base image
+FROM ubuntu:20.04
 
-RUN curl -fsSL https://ollama.com/install.sh | sh
-RUN ollama pull steamdj/llama3.1-cpu-only
+# Set environment variables
+ENV DEBIAN_FRONTEND=noninteractive
 
-# Set environment variables (if needed)
-# ENV OLLAMA_API_BASE_URL=http://0.0.0.0:$SPACE_HOST/api
-ENV OLLAMA_HOST=0.0.0.0:$SPACE_HOST
+# Update the system and install required packages
+RUN apt-get update && apt-get install -y \
+    curl \
+    sudo \
+    unzip \
+    && rm -rf /var/lib/apt/lists/*
 
-# Expose the port
+# Download and install Ollama
+RUN curl -fsSL https://ollama.ai/install.sh | bash
+
+# Set the working directory
+WORKDIR /app
+
+# Expose port 8050 for the Ollama server
 EXPOSE $SPACE_HOST
 
-# Run Ollama server and pull the model
-CMD ["ollama serve"]
+# Start the Ollama server on port 8050
+CMD ["ollama", "serve", "--port", $SPACE_HOST]