ex510 commited on
Commit
023a2b9
·
verified ·
1 Parent(s): fb4359d

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -30
main.py CHANGED
@@ -3,41 +3,13 @@ from pydantic import BaseModel
3
  from sentence_transformers import SentenceTransformer
4
  import uvicorn
5
 
6
- app = FastAPI(title="Text Embedding API (intfloat/multilingual-e5-small)")
7
 
8
  class TextRequest(BaseModel):
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("/")
 
3
  from sentence_transformers import SentenceTransformer
4
  import uvicorn
5
 
6
+ app = FastAPI(title="Text Embedding API (multilinguale5small)")
7
 
8
  class TextRequest(BaseModel):
9
  text: str
10
 
11
  # Load Model
12
+ model_id = 'multilingual‑e5‑small'
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
13
  model = SentenceTransformer(model_id)
14
 
15
  @app.get("/")