import gradio as gr

title = "DistilBERT"

description = "Gradio Demo for DistilBERT. To use it, simply add your text, or click one of the examples to load them. Read more at the links below."

article = "<p style='text-align: center'><a href='https://arxiv.org/abs/1910.01108' target='_blank'>DistilBERT, a distilled version of BERT: smaller, faster, cheaper and lighter</a></p>"

examples = [
    ['The goal of life is [MASK].']
]

gr.Interface.load("huggingface/distilbert-base-uncased", inputs=gr.inputs.Textbox(lines=5, label="Input Text"),title=title,description=description,article=article, examples=examples).launch(enable_queue=True)