from transformers.tools.base import launch_gradio_demo
from text_generator import TextGenerationTool

launch_gradio_demo(TextGenerationTool)