Update Dockerfile

Dockerfile

@@ -1,34 +1,14 @@
-FROM nvidia/cuda:12.1.1-runtime-ubuntu22.04
+# Start with the official Ollama Docker image
+FROM ollama/ollama
 
-# Set environment variables
-ENV NVIDIA_VISIBLE_DEVICES=all \
-    NVIDIA_DRIVER_CAPABILITIES=compute,utility
+# Install nginx to act as a reverse proxy
+RUN apt-get update && apt-get install -y nginx
 
-# Install dependencies
-RUN apt-get update && apt-get install -y --no-install-recommends \
-    curl \
-    ca-certificates \
-    && rm -rf /var/lib/apt/lists/*
+# Copy a custom nginx configuration file
+COPY nginx.conf /etc/nginx/nginx.conf
 
-# 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 && \
-    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' | \
-    tee /etc/apt/sources.list.d/nvidia-container-toolkit.list && \
-    apt-get update && apt-get install -y --no-install-recommends nvidia-container-toolkit
+# Expose port 7860 (required by Hugging Face Spaces)
+EXPOSE 7860
 
-# Install Ollama
-RUN curl -fsSL https://ollama.ai/install.sh | bash
-
-# Create Ollama data directory
-RUN mkdir -p /.ollama
-
-# Expose port
-EXPOSE 11434
-
-# Set working directory
-WORKDIR /root/.ollama
-
-# Start Ollama
-CMD ["ollama", "serve"]
+# Start both nginx and Ollama in the foreground
+CMD service nginx start && ollama serve