Alexandre Gazola commited on
Commit
e340bba
·
1 Parent(s): e4cbe9d

usando modelo mais avançado

Browse files
Files changed (2) hide show
  1. app.py +2 -2
  2. constants.py +2 -1
app.py CHANGED
@@ -58,8 +58,8 @@ class LangChainAgent:
58
 
59
  download_file_as_base64
60
 
61
- print("Waiting 60s before answering")
62
- time.sleep(10) # Delay for 60 seconds
63
 
64
  result = self.executor.invoke({
65
  "input": question,
 
58
 
59
  download_file_as_base64
60
 
61
+ #print("Waiting 60s before answering")
62
+ #time.sleep(10) # Delay for 60 seconds
63
 
64
  result = self.executor.invoke({
65
  "input": question,
constants.py CHANGED
@@ -1,7 +1,8 @@
1
  import os
2
 
 
3
  # MODEL = 'gemini-2.0-flash'
4
- MODEL = 'gemini-2.0-flash-exp'
5
  # MODEL = 'gemini-2.5-pro-exp-03-25'
6
 
7
  API_KEY = os.getenv("GEMINI_API_KEY")
 
1
  import os
2
 
3
+ MODEL = 'gemini-2.5-flash-preview-05-20'
4
  # MODEL = 'gemini-2.0-flash'
5
+ # MODEL = 'gemini-2.0-flash-exp'
6
  # MODEL = 'gemini-2.5-pro-exp-03-25'
7
 
8
  API_KEY = os.getenv("GEMINI_API_KEY")