Update Dockerfile

Dockerfile

@@ -1,20 +1,17 @@
 # Builder stage
 FROM ubuntu:latest
 
-RUN apt update && apt install curl -y
-
-
-RUN curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey \
-    | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg
-    
-RUN curl -s -L https://nvidia.github.io/libnvidia-container/stable/deb/nvidia-container-toolkit.list \
-    | sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' \
-    | sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list
-    
-RUN sudo apt-get update
-
-RUN apt-get install -y nvidia-container-toolkit
-
+# Install dependencies
+RUN apt-get update && apt-get install -y \
+    curl \
+    gnupg \
+    && curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg \
+    && echo "deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://nvidia.github.io/libnvidia-container/stable/deb/ $(. /etc/os-release; echo $UBUNTU_CODENAME) main" > /etc/apt/sources.list.d/nvidia-container-toolkit.list \
+    && apt-get update \
+    && apt-get install -y nvidia-container-toolkit \
+    && rm -rf /var/lib/apt/lists/*
+
+# Install ollama
 RUN curl https://ollama.ai/install.sh | sh
 
 # Create the directory and give appropriate permissions
@@ -28,10 +25,12 @@ RUN chmod +x /entrypoint.sh
 
 # Set the entry point script as the default command
 ENTRYPOINT ["/entrypoint.sh"]
+
+# Set default arguments for the entry point script
 CMD ["ollama", "serve"]
 
 # Set the model as an environment variable (this can be overridden)
 ENV model=llama2
 
 # Expose the server port
-EXPOSE 7860
+EXPOSE 7860