finish!

app.py

@@ -77,35 +77,28 @@ def conversion(source, text):
 
     return generated_text
 
-title = 'KoTAN Translator & Speech-style converter'
+title = 'KoTAN Translator & Speech-style converter. If you want to download as pip package, please visit our github. (https://github.com/KoJLabs/KoTAN)'
 lang = ['English','Korean']
 style = ['formal', 'informal', 'android', 'azae', 'chat', 'choding', 'emoticon', 'enfp', \
          'gentle', 'halbae', 'halmae', 'joongding', 'king', 'naruto', 'seonbi', 'sosim', 'translator']
 
-with gr.Blocks() as demo:
-    gr.Markdown("KoTAN: Korean Translation and Augmentation with fine-tuned NLLB. You can exercise translation tasks, and speech-style conversion. \
-                If you want to download as pip package, please visit our github. (https://github.com/KoJLabs/KoTAN)")
     
-    # with gr.Tab("Translation"):
-    #     translator_app = gr.Interface(
-    #         fn=translate,
-    #         inputs=[gr.inputs.Dropdown(choices=lang, label='Source Language'), gr.inputs.Dropdown(choices=lang, label='Target Language'), gr.inputs.Textbox(lines=5, label='Text to Translate')],
-    #         outputs=[gr.outputs.Textbox(label='Translated Text')],
-    #         title=title,
-    #         # description = 'KoTAN: Korean Translation and Augmentation with fine-tuned NLLB. If you want to download as pip package, please visit our github. (https://github.com/KoJLabs/KoTAN)',
-    #         # article='Jisu, Kim. Juhwan, Lee',
-    #         enable_queue=True,
-    #     )
+translator_app = gr.Interface(
+    fn=translate,
+    inputs=[gr.inputs.Dropdown(choices=lang, label='Source Language'), gr.inputs.Dropdown(choices=lang, label='Target Language'), gr.inputs.Textbox(lines=5, label='Text to Translate')],
+    outputs=[gr.outputs.Textbox(label='Translated Text')],
+    title=title,
+    enable_queue=True,
+)
     
-    with gr.Tab("Speech-style conversion"):
-        translator_app = gr.Interface(
-            fn=conversion,
-            inputs=[gr.inputs.Dropdown(choices=style, label='Speech Style'), gr.inputs.Textbox(lines=5, label='Text to style conversion')],
-            outputs=[gr.outputs.Textbox(label='Converted Text')],
-            title=title,
-            # description = 'KoTAN: Korean Translation and Augmentation with fine-tuned NLLB. If you want to download as pip package, please visit our github. (https://github.com/KoJLabs/KoTAN)',
-            # article='Jisu, Kim. Juhwan, Lee',
-            enable_queue=True,
-        )
+conversion_app = gr.Interface(
+    fn=conversion,
+    inputs=[gr.inputs.Dropdown(choices=style, label='Speech Style'), gr.inputs.Textbox(lines=5, label='Text to style conversion')],
+    outputs=[gr.outputs.Textbox(label='Converted Text')],
+    title=title,
+    enable_queue=True,
+)
+
+demo = gr.TabbedInterface([translator_app, conversion_app], ["Translator", "Speech Style converter"])
 
 demo.launch()