from transformers import pipeline
import gradio as gr


title = "umt5-base"
description = """
Task: Text2Text Generation\n
Trained dataset: mC4 corpus \n
Languages: 107 languages \n
Developer: Google \n
Paper: https://openreview.net/forum?id=kXwdL1cWOAi
"""
article = "<p style='text-align: center'><a href='https://huggingface.co/google/umt5-base' target='_blank'>google/umt5-base</a></p>"

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

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

demo.launch()