Upload agent

agent.json

@@ -3,12 +3,12 @@
         "final_answer"
     ],
     "model": {
-        "class": "HfApiModel",
+        "class": "LiteLLMModel",
         "data": {
-            "last_input_token_count": 3009,
-            "last_output_token_count": 226,
-            "model_id": "Qwen/Qwen2.5-Coder-32B-Instruct",
-            "provider": null
+            "last_input_token_count": null,
+            "last_output_token_count": null,
+            "model_id": "gemini/gemini-2.0-flash",
+            "api_base": null
         }
     },
     "managed_agents": {},

app.py

@@ -1,6 +1,6 @@
 import yaml
 import os
-from smolagents import GradioUI, CodeAgent, HfApiModel
+from smolagents import GradioUI, CodeAgent, LiteLLMModel
 
 # Get current directory path
 CURRENT_DIR = os.path.dirname(os.path.abspath(__file__))
@@ -9,9 +9,9 @@ from tools.final_answer import FinalAnswerTool as FinalAnswer
 
 
 
-model = HfApiModel(
-model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
-provider=None,
+model = LiteLLMModel(
+model_id='gemini/gemini-2.0-flash',
+api_base=None,
 )
 
 final_answer = FinalAnswer()