from text_to_image import TextToImageTool
import gradio as gr

tool = TextToImageTool()

def fn(*args, **kwargs):
    return tool(*args, **kwargs)

gr.Interface(
    fn=fn,
    inputs=tool_class.inputs,
    outputs=tool_class.outputs,
    title=tool_class.__name__,
    article=tool.description,
).queue(concurrency_count=5).launch()