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

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -9,7 +9,7 @@ class TextRequest(BaseModel):
9
  text: str
10
 
11
  # Load Model
12
- model_id = 'tencent/KaLM-Embedding-Gemma3-12B-2511'
13
  model = SentenceTransformer(model_id)
14
 
15
  @app.get("/")
 
9
  text: str
10
 
11
  # Load Model
12
+ model_id = 'google/embeddinggemma-300m'
13
  model = SentenceTransformer(model_id)
14
 
15
  @app.get("/")