changed embedding model to `text-embedding-004`
Browse files
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
|