import gradio as gr
from gradio.mix import Series

description = "Input Chinese language or English language text, submit and the machine will generate additional English language text."
description1 = "Input Chinese language text, submit and the machine will generate additional English language text."


translator = gr.Interface.load("huggingface/Helsinki-NLP/opus-mt-zh-en")
model1 = gr.Interface.load("huggingface/EleutherAI/gpt-j-6B")
model2 = gr.Interface.load("huggingface/ckiplab/gpt2-base-chinese")

tg_demo = gr.Series(translator, model1, inputs=gr.Textbox(lines=5, label="Input Text"), description=description)
gt_demo = gr.Series(model2, translator, inputs=gr.Textbox(lines=5, label="Input Text"), description=description1)

demo = gr.TabbedInterface([tg_demo, gt_demo], ["Chinese and English Translator to Text Generator GPT-J-6B", "Chinese Language Generator GPT-2 to Chinese to English Translator"])

if __name__ == "__main__":
    demo.launch()