Sentence trasformer for english-filipino

app.py

@@ -1,8 +1,20 @@
-from fastapi import FastAPI
+from sentence_transformers import SentenceTransformer
+from fastapi import FastAPI, HTTPException, Request
+from pydantic import BaseModel
+import torch
 
 app = FastAPI()
+model = SentenceTransformer("meedan/paraphrase-filipino-mpnet-base-v2")
 
 
-@app.get("/")
-def greet_json():
-    return {"Hello": "World!"}
+class TextInput(BaseModel):
+    text: str
+
+
+@app.post("/embed")
+def embed_text(input: TextInput):
+    if len(input.text) > 5000:
+        raise HTTPException(status_code=400, detail="Text too long")
+    with torch.no_grad():
+        embedding = model.encode([input.text])[0].tolist()
+    return {"embedding": embedding}

requirements.txt

@@ -1,2 +1,4 @@
 fastapi
-uvicorn[standard]
+uvicorn
+sentence-transformers
+torch