Update Dockerfile

Dockerfile

@@ -16,11 +16,10 @@ 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
+# Install NVIDIA container toolkit
 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
 RUN apt-get update && apt-get install -y nvidia-container-toolkit || true
 
 # Install Ollama
@@ -38,8 +37,8 @@ RUN chmod +x /entrypoint.sh
 # Copy the application script
 COPY app.py /app.py
 
-# Expose the Streamlit app port
-EXPOSE 7860
+# Expose both ports
+EXPOSE 7860 8501
 
 # Set the entry point script as the default command
 ENTRYPOINT ["/entrypoint.sh"]