Spaces:
Sleeping
Sleeping
| 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} | |