Update app.py

app.py

@@ -25,6 +25,10 @@ catering_service_tool = CateringServiceTool()
 superhero_party_theme_generator = SuperheroPartyThemeGenerator()
 final_answer = FinalAnswer()
 
+dynamicTool_TextToImage = smolagents.load_tool(
+    "m-ric/text-to-image",
+    trust_remote_code=True
+)
 
 with open(os.path.join(CURRENT_DIR, "prompts.yaml"), 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
@@ -46,7 +50,7 @@ with open(os.path.join(CURRENT_DIR, "prompts.yaml"), 'r') as stream:
 #     prompt_templates=prompt_templates
 # )
 codeAgent_Party = CodeAgent(model=InferenceClientModel(), max_steps=10, verbosity_level=2,
-    tools   =  [final_answer, web_search, visit_webpage, suggest_menu, catering_service_tool, superhero_party_theme_generator],
+    tools   =  [final_answer, web_search, visit_webpage, suggest_menu, catering_service_tool, superhero_party_theme_generator, dynamicTool_TextToImage],
     executor_type='local',
     executor_kwargs={},
     max_print_outputs_length=None,