rasyosef commited on
Commit
871e777
1 Parent(s): ae1e61f

changed embedding model to `text-embedding-004`

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -25,7 +25,7 @@ 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-004",
29
  title="wikipedia page of the 'Oppenheimer' movie",
30
  embed_batch_size=16,
31
  api_key=api_key
 
25
  def prepare_query_engine(api_key):
26
  # Gemini Embeddings
27
  embed_model = GeminiEmbedding(
28
+ model="models/text-embedding-004",
29
  title="wikipedia page of the 'Oppenheimer' movie",
30
  embed_batch_size=16,
31
  api_key=api_key