Update app.py

app.py

@@ -4,6 +4,7 @@ import src.Translate as Translate
 from typing import Optional
 from fastapi_mcp import FastApiMCP
 from huggingface_hub import hf_hub_download, list_repo_files
+from sentence_transformers import SentenceTransformer
 
 app = FastAPI()
 # app = FastAPI(docs_url="/docs")
@@ -11,6 +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"]
 
 @app.get("/")
 def index(request: Request):
@@ -110,6 +112,13 @@ def bergamot(input_text: list[str] = Query(description="Input list of strings"),
         response, message_text = error, error
     return {"input": input_text, "translated_text": response, "message_text": message_text}
 
+@app.get("/paraphrase", operation_id="get_paraphrase", description="Paraphrase text", tags=["paraphrase"], summary="Paraphrase text")
+def embed(text: str, model: str  = EMBEDDING_MODELS[0]):
+    model = SentenceTransformer("all-MiniLM-L6-v2")
+    embeddings = model.encode(sentences)
+    print(embeddings.shape)
+    return {"input": text, "embeddings": embeddings, "shape": embeddings.shape}
+
 # Create an MCP server based on this app
 mcp = FastApiMCP(
     app,