File size: 478 Bytes
79fe766
 
 
 
 
 
 
 
 
 
 
df0e186
 
 
 
79fe766
df0e186
 
79fe766
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
from fastapi import FastAPI
from pydantic import BaseModel
from sentence_transformers import SentenceTransformer

app = FastAPI()

model = SentenceTransformer('sentence-transformers/LaBSE')

class EmbedRequest(BaseModel):
    text: str

class EmbedResponse(BaseModel):
    embedding: list

@app.post("/vectorize", response_model=EmbedResponse)
async def embed_text(request: EmbedRequest):
    embedding = model.encode(request.text).tolist()
    return {"embedding": embedding}