Edenbaba commited on
Commit
88a2d34
·
verified ·
1 Parent(s): a67cd53

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -5
app.py CHANGED
@@ -103,6 +103,8 @@ image_generation_tool = Tool.from_space(
103
  description="Generate an image from a prompt"
104
  )
105
 
 
 
106
 
107
  final_answer = FinalAnswerTool()
108
  web_search = DuckDuckGoSearchTool()
@@ -122,16 +124,12 @@ model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may
122
  custom_role_conversions=None,
123
  )
124
 
125
-
126
- # Import tool from Hub
127
- image_generation_tool2 = load_tool("agents-course/text-to-image", trust_remote_code=True)
128
-
129
  with open("prompts.yaml", 'r') as stream:
130
  prompt_templates = yaml.safe_load(stream)
131
 
132
  agent = CodeAgent(
133
  model=model,
134
- tools=[final_answer, image_generation_tool, my_custom_tool, get_current_time_in_timezone, web_search, visit_webpage, python_tool, image_generation_tool,
135
  suggest_menu, catering_service_tool, SuperheroPartyThemeTool()], ## add your tools here (don't remove final answer)
136
  max_steps=10,
137
  verbosity_level=1,
 
103
  description="Generate an image from a prompt"
104
  )
105
 
106
+ # Import tool from Hub
107
+ image_generation_tool2 = load_tool("agents-course/text-to-image", trust_remote_code=True)
108
 
109
  final_answer = FinalAnswerTool()
110
  web_search = DuckDuckGoSearchTool()
 
124
  custom_role_conversions=None,
125
  )
126
 
 
 
 
 
127
  with open("prompts.yaml", 'r') as stream:
128
  prompt_templates = yaml.safe_load(stream)
129
 
130
  agent = CodeAgent(
131
  model=model,
132
+ tools=[final_answer, image_generation_tool, my_custom_tool, get_current_time_in_timezone, web_search, visit_webpage, python_tool, image_generation_tool2,
133
  suggest_menu, catering_service_tool, SuperheroPartyThemeTool()], ## add your tools here (don't remove final answer)
134
  max_steps=10,
135
  verbosity_level=1,