Update Dockerfile

Dockerfile

@@ -1,15 +1,20 @@
-# Builder stage
+# Use Ubuntu as the base image
 FROM ubuntu:latest
 
-# Update packages and install dependencies
+# Update package list and install required system dependencies
 RUN apt-get update && apt-get install -y \
     curl \
     gnupg \
     python3 \
+    python3-venv \
     python3-pip
 
-# Install required Python packages
-RUN pip3 install streamlit requests
+# Set up a virtual environment for Python
+RUN python3 -m venv /venv
+ENV PATH="/venv/bin:$PATH"
+
+# Install required Python packages inside the virtual environment
+RUN pip install --no-cache-dir streamlit requests
 
 # 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 \
@@ -21,7 +26,7 @@ RUN apt-get update && apt-get install -y nvidia-container-toolkit || true
 # Install Ollama
 RUN curl https://ollama.ai/install.sh | sh
 
-# Create the directory and give appropriate permissions
+# Create the directory and set appropriate permissions
 RUN mkdir -p /.ollama && chmod 777 /.ollama
 
 WORKDIR /.ollama