app.py

from transformers import launch_gradio_demo
from tool import TextToImageTool
import spaces

@spaces.GPU

launch_gradio_demo(TextToImageTool)