from fastapi import FastAPI
from pydantic import BaseModel
from sentence_transformers import SentenceTransformer

model = SentenceTransformer("sergeyzh/BERTA")
app = FastAPI()

class TextInput(BaseModel):
    text: str

@app.get("/ping")
def ping():
    return {"status": "ok"}

@app.post("/embed")
async def embed_text(data: TextInput):
    embedding = model.encode(data.text).tolist()
    return {"embedding": embedding}