update

app.py

@@ -11,8 +11,7 @@ from Gradio_UI import GradioUI
 
 
 @tool
-# it's import to specify the return type
-def generate_image(prompt: str, style: str) -> PIL.Image.Image:
+def generate_image(prompt: str, style: str) -> str:
     """A tool that generates an image based on a prompt and a style.
     Args:
         prompt: the prompt to generate the image
@@ -24,11 +23,11 @@ def generate_image(prompt: str, style: str) -> PIL.Image.Image:
 
         # Generate the image
         image = image_generation_tool(full_prompt)
-    
+
         # Convert the PIL image to AgentImage
         from smolagents.agent_types import AgentImage
         agent_image = AgentImage(image)
-     
+ 
         # Return the AgentImage object
         return agent_image
     except Exception as e: