Update app.py

app.py

@@ -63,31 +63,35 @@ def answer_questions(context, question):
 # App Interface
 with gr.Blocks() as app:
     gr.Markdown("<center> <h1>Question Answering with Short and Long Answer Models </h1> </center><hr>")
-    with gr.Row():
-        with gr.Column():
-            context_input = gr.Textbox(lines=7, label="Context", placeholder="Input Context here...") 
-            question_input = gr.Textbox(label="Question", placeholder="Input Question here...")
-            submit_btn = gr.Button("Submit")
-            gr.ClearButton([context_input,question_input])
-        with gr.Column():
+    with gr.Tab("QA Short Answer"):
+        with gr.Row():
+            with gr.Column():
+                context_input = gr.Textbox(lines=7, label="Context", placeholder="Input Context here...") 
+                question_input = gr.Textbox(label="Question", placeholder="Input Question here...")
+                submit_btn = gr.Button("Submit")
+                gr.ClearButton([context_input,question_input])
+            with gr.Column():
+                with gr.Row():
+                    with gr.Column(scale=6):
+                        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=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")
+        with gr.Tab("Long Answer Prediction"):
             with gr.Row():
-                with gr.Column(scale=6):
-                    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=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])
+                with gr.Column():
+                    long_question_input = gr.Textbox(label="Question", placeholder="Input Question here...")
+                    
+    submit_btn.click(fn=answer_questions, inputs=[context_input, question_input], outputs=[distilbert_base_uncased_output, distilbert_base_uncased_score_output, bert_base_uncased_output, bert_base_uncased_output, roberta_base_output, roberta_base_output])
     examples='examples'
     gr.Examples(examples,[context_input, question_input],[short_answer_output, short_score_output, long_answer_output, long_score_output],answer_questions)
 if __name__ == "__main__":