Update app.py

app.py

@@ -23,18 +23,18 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
     return "What magic will you build ?"
 # -- my tool
 @tool
-def get_image(img_description:str)-> str: #it's import to specify the return type
+def get_image(img_description:str)-> 'image': #it's import to specify the return type
     #Keep this format for the description / args / args description but feel free to modify the tool
     """A tool that generate an image from a textual description 
     Args:
         img_description: a sting containing the description of the image the user wants to get
     """
-    client = Client("agents-course/text-to-image")
-    result = client.predict(
-    		param_0="Hello!!",
-    		api_name="/predict"
-            )
-    return result
+    # client = Client("agents-course/text-to-image")
+    # result = client.predict(
+    # 		param_0="Hello!!",
+    # 		api_name="/predict"
+    #         )
+    return image_generation_tool.forward(img_description)
 # -- my tool end