ex510 commited on
Commit
79f9b67
·
verified ·
1 Parent(s): c62de64

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -3,13 +3,13 @@ from pydantic import BaseModel
3
  from sentence_transformers import SentenceTransformer
4
  import uvicorn
5
 
6
- app = FastAPI(title="Text Embedding API (KaLM-Embedding-Gemma3-12B)")
7
 
8
  class TextRequest(BaseModel):
9
  text: str
10
 
11
  # Load Model
12
- model_id = 'google/embeddinggemma-300m'
13
  model = SentenceTransformer(model_id)
14
 
15
  @app.get("/")
 
3
  from sentence_transformers import SentenceTransformer
4
  import uvicorn
5
 
6
+ app = FastAPI(title="Text Embedding API (Qwen/Qwen3-Embedding-0.6B)")
7
 
8
  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("/")