Update app.py

app.py

@@ -12,6 +12,8 @@ This model is based on an encoder-decoder T5 architecture with 1.1B parameters.
 
 For more details, please refer to our paper.
 
+Note: First inference might take time as the models are downloaded on-the-go.
+
 """
 
 
@@ -107,10 +109,11 @@ with gr.Blocks(theme="soft") as demo:
         gr.Markdown("TURNA fine-tuned on part-of-speech-tagging. Enter text to parse parts of speech and pick the model.")
         with gr.Column():
             with gr.Row():
-                pos_choice = gr.Radio(choices = ["turna_pos_imst", "turna_pos_boun"], label ="Model")
-                pos_input = gr.Textbox(label="POS Input")
-                pos_output = gr.Textbox(label="POS Output")
-                pos_submit = gr.Button()
+                with gr.Column():
+                    pos_choice = gr.Radio(choices = ["turna_pos_imst", "turna_pos_boun"], label ="Model")
+                    pos_input = gr.Textbox(label="POS Input")
+                    pos_output = gr.Textbox(label="POS Output")
+                    pos_submit = gr.Button()
                 pos_submit.click(pos, inputs=[pos_input, pos_choice], outputs=pos_output)
             pos_examples = gr.Examples(examples = ner_example, inputs = [pos_input, pos_choice], outputs=pos_output, fn=pos)
     
@@ -118,9 +121,10 @@ with gr.Blocks(theme="soft") as demo:
         gr.Markdown("TURNA fine-tuned on named entity recognition. Enter text to parse named entities and pick the model.")
         with gr.Column():
             with gr.Row():
-                ner_choice = gr.Radio(choices = ["turna_ner_wikiann", "turna_ner_milliyet"], label ="Model")
-                ner_input = gr.Textbox(label="NER Input")
-                ner_submit = gr.Button()
+                with gr.Column():
+                    ner_choice = gr.Radio(choices = ["turna_ner_wikiann", "turna_ner_milliyet"], label ="Model")
+                    ner_input = gr.Textbox(label="NER Input")
+                    ner_submit = gr.Button()
                 ner_output = gr.Textbox(label="NER Output")
                 
                 ner_submit.click(ner, inputs=[ner_input, ner_choice], outputs=ner_output)
@@ -129,20 +133,22 @@ with gr.Blocks(theme="soft") as demo:
         gr.Markdown("TURNA fine-tuned on paraphrasing. Enter text to paraphrase and pick the model.")
         with gr.Column():
             with gr.Row():
+                with gr.Column():
                 paraphrasing_choice = gr.Radio(choices = ["turna_paraphrasing_tatoeba", "turna_paraphrasing_opensubtitles"], label ="Model")
                 paraphrasing_input = gr.Textbox(label = "Paraphrasing Input")
                 paraphrasing_submit = gr.Button()
-                paraphrasing_output = gr.Text(label="Paraphrasing Output")
+            paraphrasing_output = gr.Text(label="Paraphrasing Output")
                 
-                paraphrasing_submit.click(paraphrase, inputs=[paraphrasing_input, paraphrasing_choice], outputs=paraphrasing_output)
+            paraphrasing_submit.click(paraphrase, inputs=[paraphrasing_input, paraphrasing_choice], outputs=paraphrasing_output)
             paraphrase_examples = gr.Examples(examples = long_text, inputs = [paraphrasing_input, paraphrasing_choice], outputs=paraphrasing_output,  fn=paraphrase)
     with gr.Tab("Summarization"):
         gr.Markdown("TURNA fine-tuned on summarization. Enter text to summarize and pick the model.")
         with gr.Column():
             with gr.Row():
-                sum_choice = gr.Radio(choices = ["turna_summarization_mlsum", "turna_summarization_tr_news"], label ="Model")
-                sum_input = gr.Textbox(label = "Summarization Input")
-                sum_submit = gr.Button()
+                with gr.Column():
+                    sum_choice = gr.Radio(choices = ["turna_summarization_mlsum", "turna_summarization_tr_news"], label ="Model")
+                    sum_input = gr.Textbox(label = "Summarization Input")
+                    sum_submit = gr.Button()
                 sum_output = gr.Textbox(label = "Summarization Output")
                 
                 sum_submit.click(summarize, inputs=[sum_input, sum_choice], outputs=sum_output)