adding GPU spaces function

app.py

@@ -3,10 +3,20 @@ import datetime
 import requests
 import pytz
 import yaml
+import spaces
 from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
 
+image_generation_tool = load_tool(
+    "agents-course/text-to-image",
+    trust_remote_code=True
+)
+
+@spaces.GPU
+def generate_image(prompt):
+    return image_generation_tool(prompt)
+
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
 def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return type