Update Dockerfile

Dockerfile

@@ -1,23 +1,25 @@
 # Builder stage
 FROM ubuntu:latest
 
-# Update packages and install curl and gnupg
+# Update packages and install necessary dependencies
 RUN apt-get update && apt-get install -y \
     curl \
-    gnupg
+    gnupg \
+    python3 \
+    python3-pip
 
 # Add NVIDIA package repositories
 RUN 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
 
-# Install NVIDIA container toolkit (Check for any updated methods or URLs for Ubuntu jammy)
+# Install NVIDIA container toolkit
 RUN apt-get update && apt-get install -y nvidia-container-toolkit || true
 
-# Install application
+# Install Ollama
 RUN curl https://ollama.ai/install.sh | sh
-# Below is to fix embedding bug as per
-# RUN curl -fsSL https://ollama.com/install.sh | sed 's#https://ollama.com/download#https://github.com/jmorganca/ollama/releases/download/v0.1.29#' | sh
 
+# Fix embedding bug (optional)
+# RUN curl -fsSL https://ollama.com/install.sh | sed 's#https://ollama.com/download#https://github.com/jmorganca/ollama/releases/download/v0.1.29#' | sh
 
 # Create the directory and give appropriate permissions
 RUN mkdir -p /.ollama && chmod 777 /.ollama
@@ -28,12 +30,11 @@ WORKDIR /.ollama
 COPY entrypoint.sh /entrypoint.sh
 RUN chmod +x /entrypoint.sh
 
+# Copy the application script
+COPY app.py /app.py
+
 # Set the entry point script as the default command
 ENTRYPOINT ["/entrypoint.sh"]
-CMD ["ollama", "serve"]
-
-# Set the model as an environment variable (this can be overridden)
-ENV model=${model}
 
 # Expose the server port
 EXPOSE 7860