Hugging Face's logo

Spaces:
rasyosef
/
RAG-with-Gemini-Pinecone-LlamaIndex
Running

App Files Community
rasyosef commited on
Commit
4f3e16d
•
1 Parent(s): 871e777

better retrieval

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/text-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_name="models/text-embedding-004",
29
  title="wikipedia page of the 'Oppenheimer' movie",
30
  embed_batch_size=16,
31
  api_key=api_key