Update app.py

app.py

@@ -1,22 +1,26 @@
 from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
 from tools.final_answer import FinalAnswerTool
-from Gradio_UI import GradioUI  # assuming you created this file
+from Gradio_UI import GradioUI
 
-# You must define or import image_generator
-image_generator = HfApiModel("stabilityai/stable-diffusion-2")  # Example model
+# Define your image generator tool
+image_generator = HfApiModel("stabilityai/stable-diffusion-2")
 
-# Create the agent and register tools
+# Example: Use Hugging Face API model as the LLM
+llm_model = HfApiModel("mistralai/Mixtral-8x7B-Instruct-v0.1")  # Example, you can use another model
+
+# List of tools for the agent
 tools = [
     DuckDuckGoSearchTool(),
     FinalAnswerTool(),
-    image_generator  # This acts as a tool
+    image_generator
 ]
 
-agent = CodeAgent(tools=tools)
+# Create the agent with model and tools
+agent = CodeAgent(model=llm_model, tools=tools)
 
 # Optional: You can run a test outside Gradio
-# image = image_generator(prompt="A high-quality, photorealistic image of a cat sleeping on a windowsill.")
-# final_answer(image)
+image = image_generator(prompt="A high-quality, photorealistic image of a cat sleeping on a windowsill.")
+final_answer(image)
 
-# Launch Gradio UI
-GradioUI(agent).launch()
+# Launch the Gradio UI
+GradioUI(agent).launch()