ex510 commited on
Commit
fb4359d
·
verified ·
1 Parent(s): 759db29

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +29 -1
main.py CHANGED
@@ -9,7 +9,35 @@ class TextRequest(BaseModel):
9
  text: str
10
 
11
  # Load Model
12
- model_id = 'Qwen/Qwen3-Embedding-0.6B'
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
13
  model = SentenceTransformer(model_id)
14
 
15
  @app.get("/")
 
9
  text: str
10
 
11
  # Load Model
12
+ model_id = '
13
+ class TextRequest(BaseModel):
14
+ text: str
15
+
16
+ # Load Model
17
+ model_id = 'intfloat/multilingual-e5-small'
18
+ model = SentenceTransformer(model_id)
19
+
20
+ @app.get("/")
21
+ def home():
22
+ return {"status": "online", "model": model_id, "endpoint": "/embed/text"}
23
+
24
+ @app.post("/embed/text")
25
+ async def embed_text(request: TextRequest):
26
+ try:
27
+ # Generate embedding
28
+ embedding = model.encode(request.text).tolist()
29
+
30
+ return {
31
+ "success": True,
32
+ "model": model_id,
33
+ "dimension": len(embedding),
34
+ "embedding": embedding
35
+ }
36
+ except Exception as e:
37
+ raise HTTPException(status_code=400, detail=str(e))
38
+
39
+ if __name__ == "__main__":
40
+ uvicorn.run(app, host="0.0.0.0", port=7860)'
41
  model = SentenceTransformer(model_id)
42
 
43
  @app.get("/")