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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ from llama_index.embeddings.fastembed import FastEmbedEmbedding
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"):
21
  with open(filename, "wb") as f:
 
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.1, model_name="models/gemini-pro")
19
 
20
  def write_to_file(content, filename="./files/test.pdf"):
21
  with open(filename, "wb") as f: