from transformers import pipeline
import gradio as gr


title = "umt5-xl"
description = """
Task: Text2Text Generation\n
Trained dataset: mC4 corpus \n
Languages: 107 languages \n
Model: https://huggingface.co/google/umt5-xl\n
Developer: Google \n
Paper: https://openreview.net/forum?id=kXwdL1cWOAi
"""

examples=[["I am "], ["私は"], ["我是"], ["ฉันคือ"]]

demo =gr.load(
    "huggingface/google/umt5-xl",
    inputs=gr.Textbox(lines=5, label="Input Text"),
    title=title,
    description=description,
    examples=examples,
    theme="freddyaboulton/dracula_revamped",
    )

demo.launch()