Update app.py

app.py

@@ -10,7 +10,7 @@ from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 
-llm = InferenceClient('microsoft/phi-4')
+llm = InferenceClient('Qwen/Qwen2.5-0.5B-Instruct')
 
 @tool
 def write_image_spec(pointers: str)-> str: #it's import to specify the return type
@@ -74,7 +74,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, write_image_spec], ## add your tools here (don't remove final answer)
+    tools=[final_answer, write_image_spec, image_generation_tool],
     max_steps=6,
     verbosity_level=1,
     grammar=None,