Extended to 5 gunicorn workers, fixed embedding endpoint

Dockerfile

@@ -1,43 +1,4 @@
-# FROM python
-
-# WORKDIR /code
-
-# COPY ./requirements.txt /code/requirements.txt
-
-# RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
-
-# COPY . .
-
-# CMD ["gunicorn", "-w", "1", "main:app"]
-
-# ==============================================================================
-
-# FROM python
-
-# WORKDIR /code
-
-# COPY ./requirements.txt /code/requirements.txt
-
-# RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
-
-# 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
+FROM python:3
 
 RUN useradd -m -u 1000 user
 USER user
@@ -52,4 +13,4 @@ RUN mkdir $HOME/.cache
 
 RUN pip install --no-cache-dir --upgrade -r requirements.txt
 
-CMD ["gunicorn", "-w", "1", "-b", "0.0.0.0:7860","main:app"]
+CMD ["gunicorn", "-w", "5", "-b", "0.0.0.0:7860","main:app"]

main.py

@@ -19,7 +19,7 @@ def embedding():
     message = request.get_json()
     message = model.encode(message)
 
-    return model.encode(message['text'])
+    return {"output":model.encode(message['text'].tolist())}
 
 
 # curl -v -X POST 'http://127.0.0.1:8000/completion' --header 'Content-Type: application/json' --data '{"text": "what is an apple?"}'