Update app.py

app.py

@@ -12,7 +12,7 @@ MODELS = {'enro': 'BlackKakapo/opus-mt-en-ro',
           'roen': 'BlackKakapo/opus-mt-ro-en',
           'gemma': 'Gargaz/gemma-2b-romanian-better',
           'paraphrase': 'tuner007/pegasus_paraphrase'}
-EMBEDDING_MODELS = ["all-MiniLM-L6-v2"]
+EMBEDDING_MODELS = ["all-MiniLM-L6-v2", "sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2"]
 
 @app.get("/")
 def index(request: Request):
@@ -113,7 +113,7 @@ def bergamot(input_text: list[str] = Query(description="Input list of strings"),
     return {"input": input_text, "translated_text": response, "message_text": message_text}
 
 @app.get("/embed", operation_id="get_embeddings", description="Embed text", tags=["embed"], summary="Embed text")
-def embed(text: str, model: str  = EMBEDDING_MODELS[0]):
+def embed(text: str, model: str  = EMBEDDING_MODELS[1]):
     model = SentenceTransformer(model)
     embeddings = model.encode(text)
     print(embeddings.shape)