Update text_to_image.py

text_to_image.py

@@ -1,7 +1,7 @@
 from transformers.agents.tools import Tool
 from transformers.utils import is_accelerate_available
 import torch
-
+import spaces
 
 from diffusers import DiffusionPipeline, DPMSolverMultistepScheduler
 if is_accelerate_available():
@@ -11,6 +11,7 @@ TEXT_TO_IMAGE_DESCRIPTION = (
     "This is a tool that creates an image according to a prompt."
 )
 
+@spaces.GPU
 class TextToImageTool(Tool):
     default_checkpoint = "runwayml/stable-diffusion-v1-5"
     description = TEXT_TO_IMAGE_DESCRIPTION