from transformers import launch_gradio_demo
from tool import TextToImageTool

launch_gradio_demo(TextToImageTool)