import gradio as gr

title = "QA App | BERT base finetuned on SQuAD 1.1 in Portuguese"
description = "Forneça seu próprio parágrafo e faça perguntas sobre o texto. Quão bem o modelo responde? (este aplicativo usa o modelo https://huggingface.co/pierreguillou/bert-base-cased-squad-v1.1-portuguese)"
article = f"<div style='text-align: center; font-size: 90%;'><p>Link para o modelo <a style='color:blue;' href='https://huggingface.co/pierreguillou/bert-base-cased-squad-v1.1-portuguese' target='_blank'>BERT base QA in Portuguese</a></p><p>Blog post: <a style='color:blue;' href='https://medium.com/@pierre_guillou/nlp-modelo-de-question-answering-em-qualquer-idioma-baseado-no-bert-base-estudo-de-caso-em-12093d385e78' target='_blank'>NLP | Modelo de Question Answering em qualquer idioma baseado no BERT base (estudo de caso em português)</a> (12/02/2021)</p><p>BERT QA in Portuguese App by <a style='color:blue;' href='https://www.linkedin.com/in/pierreguillou/' target='_blank'>Pierre GUILLOU</a> | <a style='color:blue;' href='https://github.com/piegu/language-models#language-models' target='_blank'>Github Repo</a></p></div>"

context = "A pandemia de COVID-19, também conhecida como pandemia de coronavírus, é uma pandemia em curso de COVID-19, uma doença respiratória aguda causada pelo coronavírus da síndrome respiratória aguda grave 2 (SARS-CoV-2). A doença foi identificada pela primeira vez em Wuhan, na província de Hubei, República Popular da China, em 1 de dezembro de 2019, mas o primeiro caso foi reportado em 31 de dezembro do mesmo ano."

question = "Quando começou a pandemia de Covid-19 no mundo?"

gr.Interface.load(
             "huggingface/pierreguillou/bert-base-cased-squad-v1.1-portuguese",
             css=".footer {display:none !important}",
             inputs=[gr.inputs.Textbox(lines=7, default=context, label="Context paragraph"), gr.inputs.Textbox(lines=2, default=question, label="Question")],
             outputs=[gr.outputs.Textbox(label="Answer"), gr.outputs.Textbox(label="Score")],
             title=title,
             description=description,
             article=article,
             ).launch()