Spaces:
Build error
Build error
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}
|