sqfoo commited on
Commit
04e04c8
·
verified ·
1 Parent(s): efdee21

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +1 -1
agent.py CHANGED
@@ -84,7 +84,7 @@ def image_caption(dir: str) -> str:
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,
 
84
  class BasicAgent:
85
  def __init__(self):
86
  model = ChatGoogleGenerativeAI(
87
+ model="gemini-2.0-flash",
88
  temperature=0,
89
  max_tokens=1024,
90
  timeout=None,