Update app.py

app.py

@@ -93,8 +93,12 @@ def get_simple_joke() -> str:
         
     except Exception as e:
         return f"Error: {str(e)}"
+@tool
+# Import tool from Hub
+image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
-
+with open("prompts.yaml", 'r') as stream:
+    prompt_templates = yaml.safe_load(stream)
 
 final_answer = FinalAnswerTool()
 
@@ -117,7 +121,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, my_custom_tool, get_current_time_in_timezone, get_eur_exchange_rate, get_simple_joke], ## add  your tools here (don't remove final answer)
+    tools=[final_answer, my_custom_tool, get_current_time_in_timezone, get_eur_exchange_rate, get_simple_joke,image_generation_tool], ## add  your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,