Update Dockerfile

Dockerfile

@@ -1,41 +1,48 @@
+# Base image with GPU support
 FROM nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04
 
+# Set env vars
 ENV DEBIAN_FRONTEND=noninteractive
 ENV PYTHONDONTWRITEBYTECODE=1
 ENV PYTHONUNBUFFERED=1
 
-# Install system dependencies as root
+# Install Python and other system dependencies
 RUN apt-get update && apt-get install -y \
     python3 python3-pip ffmpeg curl git wget sudo \
     && rm -rf /var/lib/apt/lists/*
 
-# Create Python symlink before switching user
-RUN ln -s /usr/bin/python3 /usr/bin/python
-RUN pip install --upgrade pip
+# Ensure `python` points to `python3`
+RUN ln -sf /usr/bin/python3 /usr/bin/python && pip install --upgrade pip
 
-# Create non-root user and switch
+# Add a user to avoid running everything as root (best practice)
 RUN useradd -ms /bin/bash ollama
 USER ollama
 WORKDIR /home/ollama
 
-# Set up Python
-RUN ln -s /usr/bin/python3 /usr/bin/python
-RUN pip install --upgrade pip
-
-# Install Ollama (under user home, not root)
+# Install Ollama as user
 RUN curl -fsSL https://ollama.com/install.sh | sh
 
-# Set up app directory under user space
-WORKDIR /home/ollama/app
-COPY requirements.txt .
-RUN pip install -r requirements.txt
+# Switch back to root to install Python packages globally
+USER root
+
+# Create working directory for your app
+WORKDIR /app
+
+# Copy project files
 COPY . .
 
-# Download Hugging Face interface file
+# Download SpeechBrain model interface if needed
 RUN wget -O src/custome_interface.py https://huggingface.co/Jzuluaga/accent-id-commonaccent_xlsr-en-english/resolve/main/custom_interface.py
 
-# Expose port for Streamlit
+# Install Python dependencies
+RUN pip install -r requirements.txt
+
+# Expose Streamlit default port
 EXPOSE 8501
 
-# Start Ollama and the app
-CMD bash -c "ollama serve & sleep 5 && ollama pull gemma3 && streamlit run streamlit_app.py --server.port=8501 --server.address=0.0.0.0"
+# Entrypoint: Start Ollama server, wait a bit, pull model, then launch Streamlit
+CMD bash -c "\
+    /home/ollama/.ollama/bin/ollama serve & \
+    sleep 5 && \
+    /home/ollama/.ollama/bin/ollama pull gemma3 && \
+    streamlit run streamlit_app.py --server.port=8501 --server.address=0.0.0.0"