Update app.py

app.py

@@ -38,14 +38,12 @@ def get_current_time_in_timezone(timezone: str) -> str:
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
 @tool
-def generate_image(prompt:str, num_images: int=1, size: str="512x512") -> list:
+def generate_image(prompt:str) -> list:
     """Generate image(s) from a text prompt via HF text-to-image.
     Args:
         prompt: description of the image
-        num_images: how many to produce
-        size: image resolution (e.g. "512x512")
     """
-    return image_generation_tool(prompt=prompt, num_images=num_images, size=size)    
+    return image_generation_tool(prompt)    
 
 @tool
 def duckduckgo_search(query: str, max_results: int=5) -> str: