ex510 commited on
Commit
4e20d48
·
verified ·
1 Parent(s): a487724

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -5,14 +5,14 @@ import uvicorn
5
  import asyncio
6
  from concurrent.futures import ThreadPoolExecutor
7
 
8
- app = FastAPI(title="Text Embedding API (multilingual-e5-small)")
9
 
10
  class TextRequest(BaseModel):
11
  text: str = Field(..., min_length=1, max_length=10000, description="Text to embed")
12
 
13
  # Globals
14
  model = None
15
- model_id = 'intfloat/multilingual-e5-small'
16
  executor = ThreadPoolExecutor(max_workers=4)
17
 
18
  @app.on_event("startup")
 
5
  import asyncio
6
  from concurrent.futures import ThreadPoolExecutor
7
 
8
+ app = FastAPI(title="Text Embedding API (Qwen/Qwen3-Embedding-0.6B)")
9
 
10
  class TextRequest(BaseModel):
11
  text: str = Field(..., min_length=1, max_length=10000, description="Text to embed")
12
 
13
  # Globals
14
  model = None
15
+ model_id = 'Qwen/Qwen3-Embedding-0.6B'
16
  executor = ThreadPoolExecutor(max_workers=4)
17
 
18
  @app.on_event("startup")