sqfoo commited on
Commit
efdee21
·
verified ·
1 Parent(s): 28040a7

update model setting

Browse files
Files changed (1) hide show
  1. agent.py +2 -2
agent.py CHANGED
@@ -84,9 +84,9 @@ def image_caption(dir: str) -> str:
84
  class BasicAgent:
85
  def __init__(self):
86
  model = ChatGoogleGenerativeAI(
87
- model="gemini-2.0-flash",
88
  temperature=0,
89
- max_tokens=None,
90
  timeout=None,
91
  max_retries=2,
92
  google_api_key="AIzaSyAxVUPaGJIgdxB46ZR0RWPKSjB9a63Z80o",
 
84
  class BasicAgent:
85
  def __init__(self):
86
  model = ChatGoogleGenerativeAI(
87
+ model="gemini-2.0-flash-lite",
88
  temperature=0,
89
+ max_tokens=1024,
90
  timeout=None,
91
  max_retries=2,
92
  google_api_key="AIzaSyAxVUPaGJIgdxB46ZR0RWPKSjB9a63Z80o",