modified dockerfile, kept env of transformers inside main.py -- attempt 1

Dockerfile

@@ -10,25 +10,46 @@
 
 # CMD ["gunicorn", "-w", "1", "main:app"]
 
-FROM python
+# ==============================================================================
 
-WORKDIR /code
+# FROM python
 
-COPY ./requirements.txt /code/requirements.txt
+# WORKDIR /code
 
-RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+# COPY ./requirements.txt /code/requirements.txt
 
-RUN useradd -m -u 1000 user
+# RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
 
-USER user
+# RUN useradd -m -u 1000 user
+
+# USER user
+
+# ENV HOME=/home/user \
+# 	PATH=/home/user/.local/bin:$PATH
+
+# WORKDIR $HOME/app
 
+# RUN mkdir $HOME/.cache
+
+# COPY --chown=user . $HOME/app
+
+# CMD ["gunicorn", "-w", "1", "main:app"]
+
+# ==============================================================================
+
+FROM python
+
+RUN useradd -m -u 1000 user
+USER user
 ENV HOME=/home/user \
-	PATH=/home/user/.local/bin:$PATH
+    PATH=/home/user/.local/bin:$PATH
+
+COPY --chown=user . $HOME/embeddingAPI
 
-WORKDIR $HOME/app
+WORKDIR $HOME/embeddingAPI
 
 RUN mkdir $HOME/.cache
 
-COPY --chown=user . $HOME/app
+RUN pip install --no-cache-dir --upgrade -r requirements.txt
 
 CMD ["gunicorn", "-w", "1", "main:app"]

main.py

@@ -2,8 +2,11 @@ from flask import Flask
 from flask import request
 from sentence_transformers import SentenceTransformer
 
+import os
+
 app = Flask(__name__)
 model = SentenceTransformer('embedding_model', trust_remote_code=True)
+os.environ["TRANSFORMERS_CACHE"] = "/.cache"
 
 @app.route("/embedding", methods=['POST'])
 def embedding():