from transformers.agents.tools import launch_gradio_demo
from text_to_image import TextToImageTool
import spaces

@spaces.GPU
launch_gradio_demo(TextToImageTool)