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

launch_gradio_demo(TextGenerationTool, debug=True, share=True)