Update app.py

app.py

@@ -50,13 +50,15 @@ def remove_parentheses(text):
 # Load short model
 distilbert_base_uncased = pipeline(model="Nighter/QA_wiki_data_short_answer", from_tf=True)
 bert_base_uncased  = pipeline(model="Nighter/QA_bert_base_uncased_wiki_data_short_answer", from_tf=True)
+roberta_base = pipeline(model="Nighter/QA_wiki_data_roberta_base_short_answer", from_tf=True)
 
 # Function to answer on all models
 def answer_questions(context, question):
     # long_score, long_answer = predict_answer(context, question)
     distilbert_base_uncased_result = distilbert_base_uncased(question=question, context=remove_parentheses(context))
-    bert_base_uncased_result =bert_base_uncased(question=question, context=remove_parentheses(context))
-    return distilbert_base_uncased_result['answer'], distilbert_base_uncased_result['score'], bert_base_uncased_result['answer'], bert_base_uncased_result['score'] #, long_answer, long_score
+    bert_base_uncased_result = bert_base_uncased(question=question, context=remove_parentheses(context))
+    roberta_base_result = roberta_base(question=question, context=remove_parentheses(context))
+    return distilbert_base_uncased_result['answer'], distilbert_base_uncased_result['score'], bert_base_uncased_result['answer'], bert_base_uncased_result['score'], roberta_base_result['answer'], roberta_base_result['score'] #, long_answer, long_score
 
 # App Interface
 with gr.Blocks() as app:
@@ -70,14 +72,19 @@ with gr.Blocks() as app:
         with gr.Column():
             with gr.Row():
                 with gr.Column(scale=6):
-                    short_answer_output = gr.Textbox(lines=5, label="Distil BERT Base Uncased")
+                    short_answer_output = gr.Textbox(lines=3, label="Distil BERT Base Uncased")
                 with gr.Column(scale=2):
                     short_score_output = gr.Number(label="Distil BERT Base Uncased Score")
             with gr.Row():
                 with gr.Column(scale=6):
-                    long_answer_output = gr.Textbox(lines=5, label="BERT Base Uncased")
+                    long_answer_output = gr.Textbox(lines=3, label="BERT Base Uncased")
                 with gr.Column(scale=2):
                     long_score_output = gr.Number(label="BERT Base Uncased Score")
+            with gr.Row():
+                with gr.Column(scale=6):
+                    long_answer_output = gr.Textbox(lines=3, label="RoBERTa Base")
+                with gr.Column(scale=2):
+                    long_score_output = gr.Number(label="RoBERTa Base Score")
                 
     submit_btn.click(fn=answer_questions, inputs=[context_input, question_input], outputs=[short_answer_output, short_score_output, long_answer_output, long_score_output])
     examples='examples'