Update Dockerfile

Dockerfile

@@ -1,36 +1,26 @@
-# Use Ubuntu as base image
-FROM ubuntu:22.04
+# Dockerfile for Hugging Face Spaces
+FROM ollama/ollama:latest
 
-# Set environment variables
-ENV DEBIAN_FRONTEND=noninteractive
+# Set environment variables for Hugging Face Spaces
 ENV OLLAMA_HOST=0.0.0.0
 ENV OLLAMA_PORT=7860
 
-# Update system and install dependencies
-RUN apt-get update && apt-get install -y \
-    curl \
-    ca-certificates \
-    && rm -rf /var/lib/apt/lists/*
-
-# Install Ollama
-RUN curl -fsSL https://ollama.com/install.sh | sh
-
-# Create a non-root user for running Ollama
-RUN useradd -m -s /bin/bash ollama
-
-# Create directory for Ollama data
-RUN mkdir -p /home/ollama/.ollama && chown -R ollama:ollama /home/ollama/.ollama
-
-# Switch to ollama user
-USER ollama
-WORKDIR /home/ollama
-
-# Expose the default Ollama port
-EXPOSE 11434
-
-# Health check
-HEALTHCHECK --interval=30s --timeout=10s --start-period=5s --retries=3 \
-    CMD curl -f http://localhost:11434/api/tags || exit 1
-
-# Start Ollama server
-CMD ["ollama", "serve"]
+# Hugging Face Spaces uses port 7860
+EXPOSE 7860
+
+# Create startup script
+RUN echo '#!/bin/bash\n\
+# Start Ollama server in background\n\
+ollama serve &\n\
+\n\
+# Wait for Ollama to be ready\n\
+sleep 5\n\
+\n\
+# Pull a lightweight model (you can change this)\n\
+ollama pull llama2:7b-chat\n\
+\n\
+# Keep the container running\n\
+wait' > /start.sh && chmod +x /start.sh
+
+# Run the startup script
+CMD ["/start.sh"]