Update Dockerfile

Dockerfile

@@ -8,6 +8,12 @@ RUN apt update && apt install -y python3 && apt install -y python3-pip
 RUN pip3 install litellm
 RUN pip3 install 'litellm[proxy]'
 
+# Create a directory for Ollama data
+RUN mkdir -p /.ollama
+RUN chmod -R 777 /.ollama
+
+WORKDIR /.ollama
+
 # Copy the entry point script
 COPY entrypoint.sh /entrypoint.sh
 RUN chmod +x /entrypoint.sh