Prat0 commited on
Commit
b988347
·
verified ·
1 Parent(s): 156aa34

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -8,11 +8,13 @@ import google.generativeai as genai
8
  import os
9
  import PyPDF2
10
  from io import BytesIO
 
11
 
12
 
13
 
14
  # Configure Google Gemini
15
- Settings.embed_model = GeminiEmbedding(api_key=os.getenv("GOOGLE_API_KEY"), model_name="models/embedding-001")
 
16
  Settings.llm = Gemini(api_key=os.getenv("GOOGLE_API_KEY"), temperature=0.8, model_name="models/gemini-pro")
17
 
18
  def write_to_file(content, filename="./files/test.pdf"):
 
8
  import os
9
  import PyPDF2
10
  from io import BytesIO
11
+ from llama_index.embeddings.fastembed import FastEmbedEmbedding
12
 
13
 
14
 
15
  # Configure Google Gemini
16
+ Settings.embed_model = FastEmbedEmbedding(model_name="BAAI/bge-small-en-v1.5")
17
+
18
  Settings.llm = Gemini(api_key=os.getenv("GOOGLE_API_KEY"), temperature=0.8, model_name="models/gemini-pro")
19
 
20
  def write_to_file(content, filename="./files/test.pdf"):