Create app.py

app.py

@@ -0,0 +1,16 @@
+from fastapi import FastAPI
+from pydantic import BaseModel
+from sentence_transformers import SentenceTransformer
+import os
+
+app = FastAPI()
+
+model = SentenceTransformer("thenlper/gte-small")
+
+class EmbedInput(BaseModel):
+    text: str
+
+@app.post("/embed")
+async def embed_text(payload: EmbedInput):
+    embedding = model.encode(payload.text, normalize_embeddings=True)
+    return {"embedding": embedding.tolist()}