import gradio as gr

title = "Reformer"

description = "Gradio Demo for Reformer Model trained on 'Crime and Punishment'. 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/2001.04451' target='_blank'>Reformer: The Efficient Transformer</a></p>"

examples = [
    ['A few months later']
]

gr.Interface.load("huggingface/google/reformer-crime-and-punishment", inputs=gr.inputs.Textbox(lines=5, label="Input Text"),title=title,description=description,article=article, examples=examples).launch(enable_queue=True)