Update app.py

app.py

@@ -97,9 +97,20 @@ model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud',#
 custom_role_conversions=None,
 )
 
-
+@tool 
+def image_generation_tool(promt: str)->any:
+    """A tool that generate image from user`s promt.
+    Args:
+        promt: description of image
+    """
+    try:
+        image = image_generation_tool(promt)
+        return image
+        
+    except Exception as e:
+        return f'We have some problem {e}'
 # Import tool from Hub
-image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+#image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)