Update app.py

app.py

@@ -51,6 +51,7 @@ def remove_parentheses(text):
 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)
+longformer_base = pipeline(model="Nighter/Fine-tuned-longformer-base")
 
 # Function to answer on all models
 def answer_questions(context, question):
@@ -58,7 +59,8 @@ def answer_questions(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))
     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
+    longformer_base_result = longformer_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'], longformer_base_result['score'], longformer_base_result['answer'], roberta_base_result['score'] #, long_answer, long_score
 
 # App Interface
 with gr.Blocks() as app:
@@ -86,10 +88,15 @@ with gr.Blocks() as app:
                         roberta_base_output = gr.Textbox(lines=3, label="RoBERTa Base")
                     with gr.Column(scale=2):
                         roberta_base_score = gr.Number(label="RoBERTa Base Score")
+                with gr.Row():
+                    with gr.Column(scale=6):
+                        longformer_base_output = gr.Textbox(lines=3, label="Longformer Base")
+                    with gr.Column(scale=2):
+                        longformer_base_score = gr.Number(label="Longformer Base Score")
 
-        submit_btn.click(fn=answer_questions, inputs=[context_input, question_input], outputs=[distilbert_base_uncased_output, distilbert_base_uncased_score, bert_base_uncased_output, bert_base_uncased_score, roberta_base_output, roberta_base_score])
+        submit_btn.click(fn=answer_questions, inputs=[context_input, question_input], outputs=[distilbert_base_uncased_output, distilbert_base_uncased_score, bert_base_uncased_output, bert_base_uncased_score, roberta_base_output, roberta_base_score, longformer_base_output, longformer_base_score])
         examples='examples'
-        gr.Examples(examples,[context_input, question_input],[distilbert_base_uncased_output, distilbert_base_uncased_score, bert_base_uncased_output, bert_base_uncased_score, roberta_base_output, roberta_base_score],answer_questions)
+        gr.Examples(examples,[context_input, question_input],[distilbert_base_uncased_output, distilbert_base_uncased_score, bert_base_uncased_output, bert_base_uncased_score, roberta_base_output, roberta_base_score, longformer_base_output, longformer_base_score],answer_questions)
         
     with gr.Tab("Long Answer Prediction"):
         with gr.Row():