orrinin commited on
Commit
7ce211d
·
verified ·
1 Parent(s): 9a6b2aa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -22,7 +22,7 @@ import uuid
22
  api_key = os.environ.get("API_KEY")
23
 
24
  llm = Gemini(api_key=api_key, model_name="models/gemini-1.5-flash-latest")
25
- gemini_embedding_model = GeminiEmbedding(api_key=api_key, model_name="models/embedding-001")
26
 
27
 
28
 
 
22
  api_key = os.environ.get("API_KEY")
23
 
24
  llm = Gemini(api_key=api_key, model_name="models/gemini-1.5-flash-latest")
25
+ gemini_embedding_model = GeminiEmbedding(api_key=api_key, model_name="models/text-embedding-004")
26
 
27
 
28