Create app.py

app.py

@@ -0,0 +1,18 @@
+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}