update Dockerfile

Dockerfile

@@ -6,8 +6,22 @@ COPY ./requirements.txt /code/requirements.txt
 
 RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
 
-COPY . .
+RUN useradd -m -u 1000 user
 
-EXPOSE 7860
+USER user
 
-CMD ["shiny", "run", "app.py", "--host", "0.0.0.0", "--port", "7860"]
+ENV HOME=/home/user \
+	PATH=/home/user/.local/bin:$PATH
+
+WORKDIR $HOME/app
+
+COPY --chown=user . $HOME/app
+
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]
+
+
+# COPY . .
+
+# EXPOSE 7860
+
+# CMD ["shiny", "run", "app.py", "--host", "0.0.0.0", "--port", "7860"]

app.py

@@ -15,8 +15,8 @@ warnings.filterwarnings('ignore')
 # Load environment variables from .env file
 load_dotenv()
 
-os.environ["TRANSFORMERS_CACHE"] = "./transformers_cache"
-os.makedirs("./transformers_cache", exist_ok=True)
+# os.environ["TRANSFORMERS_CACHE"] = "./transformers_cache"
+# os.makedirs("./transformers_cache", exist_ok=True)
 
 class PDBSearchAssistant:
     def __init__(self, model_name="google/flan-t5-large"):