Spaces:
Sleeping
Sleeping
update model setting
Browse files
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=
|
| 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",
|