Update Gradio_UI.py

Gradio_UI.py

@@ -180,27 +180,27 @@ def stream_to_gradio(
 
 
 class AgentCloner:
-    def __init__(self, agent: CodeAgent) -> None:
-        self.agent = agent
-        self.model_client = agent.model
+    def __init__(self, avalaible_tools: list[callable], prompt_templates: list[dict]) -> None:
+        self.avaliable_tools = avalaible_tools
+        self.prompt_templates = prompt_templates
 
     def clone(self):
         new_model_client = HfApiModel(
-            max_tokens=self.model_client.max_tokens,
-            temperature=self.model_client.temperature,
-            model_id=self.model_client.model_id,
-            custom_role_conversions=self.model_client.custom_role_conversions,
-        )
+            max_tokens=2096,
+            temperature=0.5,
+            model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
+            custom_role_conversions=None,
+            )
         new_agent = CodeAgent(
-            model=new_model_client,
-            tools=self.agent.tools,
-            max_steps=self.agent.max_steps,
-            verbosity_level=self.agent.verbosity_level,
-            grammar=self.agent.grammar,
-            planning_interval=self.agent.planning_interval,
-            name=self.agent.name,
-            description=self.agent.description,
-            prompt_templates=self.agent.prompt_templates
+            model=model,
+            tools=self.avalaible_tools,
+            max_steps=6,
+            verbosity_level=1,
+            grammar=None,
+            planning_interval=None,
+            name=None,
+            description=None,
+            prompt_templates=self.prompt_templates
         )
         return new_agent
 
@@ -209,12 +209,12 @@ class AgentCloner:
 class GradioUI:
     """A one-line interface to launch your agent in Gradio"""
 
-    def __init__(self, agent: MultiStepAgent, file_upload_folder: str | None = None):
+    def __init__(self, agent_cloner: AgentCloner, file_upload_folder: str | None = None):
         if not _is_package_available("gradio"):
             raise ModuleNotFoundError(
                 "Please install 'gradio' extra to use the GradioUI: `pip install 'smolagents[gradio]'`"
             )
-        self.agent_cloner = AgentCloner(agent)
+        self.agent_cloner = AgentCloner(avalaible_tools)
         self.file_upload_folder = file_upload_folder
         if self.file_upload_folder is not None:
             if not os.path.exists(file_upload_folder):