tagny commited on
Commit
6205d71
·
verified ·
1 Parent(s): 46b2714

Update Gradio_UI.py

Browse files
Files changed (1) hide show
  1. Gradio_UI.py +4 -1
Gradio_UI.py CHANGED
@@ -25,6 +25,7 @@ from smolagents.agents import ActionStep, MultiStepAgent
25
  from smolagents.memory import MemoryStep
26
  from smolagents.utils import _is_package_available
27
  from smolagents import CodeAgent, HfApiModel
 
28
 
29
  import gradio as gr
30
 
@@ -191,9 +192,11 @@ class AgentCloner:
191
  model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
192
  custom_role_conversions=None,
193
  )
 
194
  new_agent = CodeAgent(
195
  model=new_model_client,
196
- tools=self.avalaible_tools,
 
197
  max_steps=6,
198
  verbosity_level=1,
199
  grammar=None,
 
25
  from smolagents.memory import MemoryStep
26
  from smolagents.utils import _is_package_available
27
  from smolagents import CodeAgent, HfApiModel
28
+ from tools.final_answer import FinalAnswerTool
29
 
30
  import gradio as gr
31
 
 
192
  model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
193
  custom_role_conversions=None,
194
  )
195
+ final_answer = FinalAnswerTool()
196
  new_agent = CodeAgent(
197
  model=new_model_client,
198
+ # tools=self.avalaible_tools,
199
+ tools=[final_answer],
200
  max_steps=6,
201
  verbosity_level=1,
202
  grammar=None,