Final_Assignment_Template

Sleeping

dlaima commited on Jun 5

Commit

7de7b09

verified ·

1 Parent(s): 7d83bde

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -50,7 +50,7 @@ class GeminiFlashModel:
 # Agent using Gemini
 class MyAgent:
     def __init__(self):
-        self.model = GeminiFlashModel(model_name="gemini-1.5-flash")
         self.agent = CodeAgent(tools=[DuckDuckGoSearchTool()], model=self.model)
     def __call__(self, question: str) -> str:

 # Agent using Gemini
 class MyAgent:
     def __init__(self):
+        self.model = GeminiFlashModel(model_id="gemini-1.5-flash")
         self.agent = CodeAgent(tools=[DuckDuckGoSearchTool()], model=self.model)
     def __call__(self, question: str) -> str: