placingholocaust/vector-endpoint

1from fastapi import FastAPI2from pydantic import BaseModel3from sentence_transformers import SentenceTransformer4 5app = FastAPI()6 7model = SentenceTransformer('sentence-transformers/LaBSE')8 9class EmbedRequest(BaseModel):10    text: str11 12class EmbedResponse(BaseModel):13    embedding: list14 15@app.post("/vectorize", response_model=EmbedResponse)16async def embed_text(request: EmbedRequest):17    embedding = model.encode(request.text).tolist()18    return {"embedding": embedding}19 20