File size: 581 Bytes
6b4360c 809de12 6b4360c e648cf1 809de12 e648cf1 6b4360c 809de12 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
from sentence_transformers import SentenceTransformer
from fastapi import FastAPI
from pydantic import BaseModel
import torch
app = FastAPI()
model_name = "meedan/paraphrase-filipino-mpnet-base-v2"
model = SentenceTransformer(model_name)
class TextInput(BaseModel):
text: str
@app.post("/embed")
def embed_text(input: TextInput):
with torch.no_grad():
embedding = model.encode([input.text])[0].tolist()
return {"embedding": embedding}
@app.get("/")
def read_root():
return f"Encode text to numerical vectors using sentence transformer: {model_name}"
|