rasyosef commited on
Commit
bee07a3
1 Parent(s): 029d0f7

updated embedding model

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -25,8 +25,8 @@ def is_valid_gemini_api_key(api_key):
25
  def prepare_query_engine(api_key):
26
  # Gemini Embeddings
27
  embed_model = GeminiEmbedding(
28
- model="models/embedding-001",
29
- title="Oppenheimer movie wikipedia",
30
  embed_batch_size=16,
31
  api_key=api_key
32
  )
 
25
  def prepare_query_engine(api_key):
26
  # Gemini Embeddings
27
  embed_model = GeminiEmbedding(
28
+ model="models/embedding-004",
29
+ title="wikipedia page of the 'Oppenheimer' movie",
30
  embed_batch_size=16,
31
  api_key=api_key
32
  )