Update text_to_image.py

text_to_image.py

@@ -1,6 +1,8 @@
 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():
@@ -10,7 +12,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