Update Dockerfile

Dockerfile

@@ -14,13 +14,15 @@
 
 FROM python:3.10.9
 
-USER root
-
 WORKDIR /code
 
-RUN mkdir /code/cache \
-    && chown -R root:root /code/cache \
-    && chmod -R 777 /code/cache
+RUN apt-get update && apt-get install -y python3-venv
+
+RUN python3 -m venv /venv
+
+ENV PATH="/venv/bin:$PATH"
+
+RUN mkdir /code/cache
 
 COPY ./requirements.txt /code/requirements.txt
 
@@ -29,3 +31,4 @@ RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
 COPY . .
 
 CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]
+