Update app.py

app.py

@@ -51,7 +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")
+longformer_base = pipeline(model="aware-ai/longformer-squadv2")
 
 # Function to answer on all models
 def answer_questions(context, question):
@@ -74,22 +74,22 @@ with gr.Blocks() as app:
                 gr.ClearButton([context_input,question_input])
             with gr.Column():
                 with gr.Row():
-                    with gr.Column(scale=6):
+                    with gr.Column(scale=4):
                         distilbert_base_uncased_output = gr.Textbox(lines=3, label="Distil BERT Base Uncased")
                     with gr.Column(scale=2):
                         distilbert_base_uncased_score = gr.Number(label="Distil BERT Base Uncased Score")
                 with gr.Row():
-                    with gr.Column(scale=6):
+                    with gr.Column(scale=4):
                         bert_base_uncased_output = gr.Textbox(lines=3, label="BERT Base Uncased")
                     with gr.Column(scale=2):
                         bert_base_uncased_score = gr.Number(label="BERT Base Uncased Score")
                 with gr.Row():
-                    with gr.Column(scale=6):
+                    with gr.Column(scale=4):
                         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):
+                    with gr.Column(scale=4):
                         longformer_base_output = gr.Textbox(lines=3, label="Longformer Base")
                     with gr.Column(scale=2):
                         longformer_base_score = gr.Number(label="Longformer Base Score")