updated embedding model
Browse files
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-
|
29 |
-
title="Oppenheimer movie
|
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 |
)
|