Update Dockerfile

Dockerfile

@@ -2,8 +2,9 @@ FROM ubuntu:22.04
 
 # Set environment variables
 ENV DEBIAN_FRONTEND=noninteractive
+ENV HOME=/root
 
-# Install dependencies, code-server, and ollama without sudo in the initial list
+# Install dependencies, code-server, and Ollama
 RUN apt-get update && \
     apt-get install -y curl wget gpg apt-transport-https git nodejs npm python3 python3-pip && \
     curl -fsSL https://code-server.dev/install.sh | sh && \
@@ -13,13 +14,13 @@ RUN apt-get update && \
 # Create a directory for the workspace
 RUN mkdir -p /workspace
 
-# Create configuration directory
-RUN mkdir -p /root/.config/code-server
+# Create configuration directory for code-server and Ollama
+RUN mkdir -p /root/.config/code-server /root/.ollama
 
 # Configure code-server to run on port 7860 (Hugging Face Spaces default)
 RUN echo "bind-addr: 0.0.0.0:7860\nauth: none\ncert: false" > /root/.config/code-server/config.yaml
 
-# Install ollama after code-server is set up
+# Install Ollama after code-server is set up
 RUN curl -fsSL https://ollama.com/install.sh | sh || true
 
 # Install some useful VS Code extensions
@@ -29,10 +30,10 @@ RUN code-server --install-extension ms-python.python && \
 
 # Create a startup script
 RUN echo '#!/bin/bash\n\
-# Start ollama in the background\n\
+# Start Ollama in the background\n\
 /usr/local/bin/ollama serve &\n\
 \n\
-# Give ollama a moment to start\n\
+# Give Ollama a moment to start\n\
 sleep 2\n\
 \n\
 # Start code-server in the foreground\n\
@@ -47,4 +48,4 @@ EXPOSE 7860 11434
 WORKDIR /workspace
 
 # Start both services
-CMD ["/start.sh"]
+CMD ["/start.sh"]