Spaces:
Build error
Build error
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 | |
async def embed_text(request: EmbedRequest): | |
embedding = model.encode(request.text).tolist() | |
return {"embedding": embedding} | |