Update Dockerfile

Dockerfile

@@ -1,13 +1,22 @@
-FROM python:3.9
+FROM python:3.9-slim
 
 RUN useradd -m -u 1000 user
 
 WORKDIR /app
 
-COPY --chown=user ./requirements.txt requirements.txt
-
+COPY requirements.txt requirements.txt
 RUN pip install --no-cache-dir --upgrade -r requirements.txt
 
-COPY --chown=user . /app
+# Set the cache directory and pre-download the model
+RUN mkdir -p /app/.cache/huggingface
+ENV TRANSFORMERS_CACHE=/app/.cache/huggingface
+
+COPY . .
+
+RUN chown -R user:user /app
+
+USER user
+
+EXPOSE 5000
 
-CMD ["gunicorn", "-b", "0.0.0.0:5000", "app:app"]
+CMD ["gunicorn", "-b", "0.0.0.0:5000", "app:app"]