Update Dockerfile

Dockerfile

@@ -1,16 +1,21 @@
-FROM python:3.9-slim
+FROM python:3.9
+
+RUN useradd -m -u 1000 user
+
+USER user
+
+ENV PATH="/home/user/.local/bin:$PATH"
 
 WORKDIR /app
 
-RUN apt-get update && apt-get install -y wget
+# Download YUGO model
+RUN wget https://huggingface.co/cigol123/YUGO-GPT-Q4_0-GGUF/resolve/main/yugogpt-q4_0.gguf
 
-ARG HF_TOKEN
-ENV HF_TOKEN=${HF_TOKEN}
-RUN wget https://huggingface.co/cigol123/YUGO-GPT-Q4_0-GGUF/resolve/main/yugogpt-q4_0.gguf --header="Authorization: Bearer ${HF_TOKEN}"
+COPY --chown=user ./requirements.txt requirements.txt
 
-COPY requirements.txt .
-RUN pip install -r requirements.txt
+RUN pip install --no-cache-dir --upgrade -r requirements.txt
 
-COPY . .
+COPY --chown=user . /app
 
 CMD ["python", "app.py"]
+