Update Dockerfile

Dockerfile

@@ -1,33 +1,19 @@
-# Use the latest version of the ollama image
 FROM ollama/ollama:latest
 
-# Install curl and create a user
-RUN apt-get update && \
-    apt-get install -y curl && \
-    apt-get clean && \
-    rm -rf /var/lib/apt/lists/* && \
-    useradd -m -u 1000 user
+RUN apt-get update && apt-get install curl -y
 
-# Set environment variables
+# https://huggingface.co/docs/hub/spaces-sdks-docker-first-demo
+RUN useradd -m -u 1000 user
+USER user
 ENV HOME=/home/user \
-    PATH=/home/user/.local/bin:$PATH \
+	PATH=/home/user/.local/bin:$PATH \
     OLLAMA_HOST=0.0.0.0 \
     OLLAMA_ORIGINS=*
 
-# Set the working directory and switch to the user
 WORKDIR $HOME/app
-USER user
 
-# Copy the Modelfile to the working directory
 COPY --chown=user:user Modelfile $HOME/app/
 
-# Pull the necessary models and start the Ollama server
-RUN ollama serve & \
-    sleep 5 && \
-    ollama pull llama3.2 && \
-    ollama pull llama3.2:1b && \
-    ollama pull granite3-moe && \
-    ollama pull granite3-moe:1b
+RUN ollama serve & sleep 5 && ollama pull qwen2:0.5b
 
-# Expose the port for the Ollama server
 EXPOSE 11434