Spaces:
Paused
Paused
Update Gradio_UI.py
Browse files- Gradio_UI.py +1 -30
Gradio_UI.py
CHANGED
@@ -180,39 +180,10 @@ def stream_to_gradio(
|
|
180 |
yield gr.ChatMessage(role="assistant", content=f"**Final answer:** {str(final_answer)}")
|
181 |
|
182 |
|
183 |
-
class AgentCloner:
|
184 |
-
def __init__(self, avalaible_tools: list[callable], prompt_templates: list[dict]) -> None:
|
185 |
-
self.avalaible_tools = avalaible_tools
|
186 |
-
self.prompt_templates = prompt_templates
|
187 |
-
|
188 |
-
def clone(self):
|
189 |
-
new_model_client = HfApiModel(
|
190 |
-
max_tokens=2096,
|
191 |
-
temperature=0.5,
|
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,
|
203 |
-
planning_interval=None,
|
204 |
-
name=None,
|
205 |
-
description=None,
|
206 |
-
prompt_templates=self.prompt_templates
|
207 |
-
)
|
208 |
-
return new_agent
|
209 |
-
|
210 |
-
|
211 |
-
|
212 |
class GradioUI:
|
213 |
"""A one-line interface to launch your agent in Gradio"""
|
214 |
|
215 |
-
def __init__(self, agent_cloner
|
216 |
if not _is_package_available("gradio"):
|
217 |
raise ModuleNotFoundError(
|
218 |
"Please install 'gradio' extra to use the GradioUI: `pip install 'smolagents[gradio]'`"
|
|
|
180 |
yield gr.ChatMessage(role="assistant", content=f"**Final answer:** {str(final_answer)}")
|
181 |
|
182 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
183 |
class GradioUI:
|
184 |
"""A one-line interface to launch your agent in Gradio"""
|
185 |
|
186 |
+
def __init__(self, agent_cloner, file_upload_folder: str | None = None):
|
187 |
if not _is_package_available("gradio"):
|
188 |
raise ModuleNotFoundError(
|
189 |
"Please install 'gradio' extra to use the GradioUI: `pip install 'smolagents[gradio]'`"
|