Update app.py

app.py

@@ -1,9 +1,10 @@
-from transformers import load_tool, ReactCodeAgent, HfApiEngine
+#from transformers import load_tool, ReactCodeAgent, HfApiEngine
 from PIL import Image, ImageDraw, ImageFont
 import tempfile
 import gradio as gr
-#from smolagents import CodeAgent, InferenceClientModel
-#from smolagents import HfApiEngine
+from smolagents import CodeAgent, InferenceClientModel
+from smolagents import HfApiEngine
+from transformers import load_tool
 
 #%% Methods
 # Function to add a label to an image
@@ -113,7 +114,8 @@ search_tool = DuckDuckGoSearchTool()
 llm_engine = HfApiEngine("Qwen/Qwen2.5-72B-Instruct")
 
 # Initialize the agent with both tools
-agent = ReactCodeAgent(tools=[image_generation_tool, search_tool], llm_engine=llm_engine)
+#agent = ReactCodeAgent(tools=[image_generation_tool, search_tool], llm_engine=llm_engine)
+agent = CodeAgent(tools=[image_generation_tool, search_tool], llm_engine=llm_engine)
 
 # Gradio interface
 def create_gradio_interface():