import gradio as gr

title = "ALBERT"

description = "Gradio Demo for ALBERT. 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/1909.11942' target='_blank'>ALBERT: A Lite BERT for Self-supervised Learning of Language Representations</a></p>"

examples = [
    ['Paris is the [MASK] of France.']
]

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