Update Dockerfile

Dockerfile

@@ -6,26 +6,32 @@ ENV PYTHONUNBUFFERED=1
 
 # Install Python and dependencies
 RUN apt-get update && apt-get install -y \
-    python3 python3-pip ffmpeg curl git wget \
+    python3 python3-pip ffmpeg curl git wget sudo \
     && rm -rf /var/lib/apt/lists/*
 
-RUN ln -s /usr/bin/python3 /usr/bin/python && pip install --upgrade pip
+# Create a non-root user
+RUN useradd -ms /bin/bash ollama
+USER ollama
+WORKDIR /home/ollama
 
-# Install Ollama
-RUN curl -fsSL https://ollama.com/install.sh | sh
+# Set up Python
+RUN ln -s /usr/bin/python3 /usr/bin/python
+RUN pip install --upgrade pip
 
-# Create app directory
-WORKDIR /app
+# Install Ollama (under user home, not root)
+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
-
 COPY . .
 
-# Ensure the custom interface is in place
+# Download Hugging Face interface file
 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
 EXPOSE 8501
 
-# Start Ollama and your app together
+# 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"