roberta qa test

app.py

@@ -1,4 +1,26 @@
 import gradio as gr
+from transformers import AutoModelForQuestionAnswering, AutoTokenizer, pipeline
+
+model_name = "deepset/roberta-base-squad2"
+
+context = "Escolha não fazer citologia cérvico-vaginal (teste de Papanicolau) ou rastreio para HPV em mulheres já submetidas a histerectomia (ou a remoção do colo uterino- traquelectomia), por outras razões que não sejam lesão pavimentosa intraepitelial de alto grau (HSIL, CIN 2 ou 3) ou carcinoma. Justificação: O cancro vaginal após histerectomia é muito raro, mais raro ainda que o cancro da mama no homem, para o qual o rastreio não está recomendado. O rastreio nestas mulheres vai mais provavelmente detetar lesões benignas que levam a testes invasivos de imediato e não à prevenção do cancro. O seguimento com citologia cérvico-vaginal está recomendado para mulheres pós-histerectomia quando esta tenha sido realizada por displasia de alto grau ou cancro, atendendo a que nestes casos o risco de cancro vaginal permanec e elevado. A citologia vaginal também pode estar indicada na presença de cancro vulvar associado ao HPV."
+
+model = AutoModelForQuestionAnswering.from_pretrained(model_name)
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+
+def qanda(question):
+    nlp = pipeline('question-answering', model=model_name, tokenizer=model_name)
+    QA_input = {
+        'question': question,
+        'context': context
+    }
+    res = nlp(QA_input)    
+
+app = gr.Interface(fn=qanda, inputs="text", outputs="text")
+
+app.launch()
+
+'''
 from transformers import pipeline, set_seed
 from transformers import BioGptTokenizer, BioGptForCausalLM
 
@@ -14,4 +36,4 @@ def biogpt_promot(inpt):
 
 app = gr.Interface(fn=biogpt_promot, inputs="text", outputs="text")
 
-app.launch()
+'''