JohnKouf commited on
Commit
a56b4bc
·
verified ·
1 Parent(s): 15eab7c

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +29 -0
app.py ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from fastapi import FastAPI, Request
2
+ from pydantic import BaseModel
3
+ from sentence_transformers import SentenceTransformer
4
+ from typing import List
5
+ import uvicorn
6
+
7
+ app = FastAPI()
8
+
9
+ # Load model once on startup
10
+ model = SentenceTransformer('lighteternal/stsb-xlm-r-greek-transfer')
11
+
12
+ # Request schema
13
+ class EmbeddingRequest(BaseModel):
14
+ texts: List[str]
15
+
16
+ # Health check endpoint
17
+ @app.get("/")
18
+ async def root():
19
+ return {"message": "Embedding API is running."}
20
+
21
+ # Embedding endpoint
22
+ @app.post("/embed")
23
+ async def embed_text(request: EmbeddingRequest):
24
+ embeddings = model.encode(request.texts).tolist()
25
+ return {"embeddings": embeddings}
26
+
27
+ # Optional for local testing
28
+ if __name__ == "__main__":
29
+ uvicorn.run(app, host="0.0.0.0", port=7860)