Upload Dockerfile with huggingface_hub

Dockerfile

@@ -6,7 +6,7 @@
 
 # Want to help us make this template better? Share your feedback here: https://forms.gle/ybq9Krt8jtBL3iCk7
 
-ARG PYTHON_VERSION=3.10.4
+ARG PYTHON_VERSION=3.10.12
 FROM python:${PYTHON_VERSION}-slim as base
 
 # Prevents Python from writing pyc files.
@@ -34,14 +34,16 @@ RUN adduser \
 # Leverage a cache mount to /root/.cache/pip to speed up subsequent builds.
 # Leverage a bind mount to requirements.txt to avoid having to copy them into
 # into this layer.
+RUN --mount=type=cache,target=/root/.cache/pip \
+    python -m pip install torch==2.3.0 --index-url https://download.pytorch.org/whl/cpu
+
 RUN --mount=type=cache,target=/root/.cache/pip \
     --mount=type=bind,source=requirements.txt,target=requirements.txt \
     python -m pip install -r requirements.txt
 
-# Switch to the non-privileged user to run the application.
-USER appuser
 
 FROM base as dev
+# ENV PYTHONPATH="${PYTHONPATH}:./api"
 ENV FLASK_APP=run.py
 ENV FLASK_DEBUG=true
 COPY . .
@@ -50,11 +52,7 @@ CMD flask run --host=0.0.0.0
 
 
 FROM base as prod
-# Copy the source code into the container.
 COPY . .
-
-# Expose the port that the application listens on.
 EXPOSE 5000
-
-# Run the application.
-CMD gunicorn 'run:app' --bind=0.0.0.0:5000
+# Switch to the non-privileged user to run the application.
+CMD gunicorn --bind 0.0.0.0:5000 api.app:app --workers 1 --log-level info --access-logfile - --error-logfile -