Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 from transformers import pipeline
 
-# pipeline_en = pipeline(task="text2text-generation", model="beyond/genius-large")
+pipeline_en = pipeline(task="text2text-generation", model="beyond/genius-large")
 
 pipeline_zh = pipeline(task="text2text-generation", model="beyond/genius-base-chinese")
 
@@ -14,23 +14,23 @@ def predict_zh(sketch):
   return generated_text
   
  
+
+
 with gr.Blocks() as demo:
-    gr.Markdown(
-        """
-        💡GENIUS – generating text using sketches! [Visit our github repo](https://github.com/beyondguo/genius)
-        The English version use `<mask>` as the mask token.
-        """)
-
-    with gr.Row():
-        with gr.Column():
-            model_input = gr.Textbox(lines=7, placeholder='Input your sketch', label='Input')
-            with gr.Row():
-                gen = gr.Button("Generate")
-                clr = gr.Button("Clear")
-               
-        outputs = gr.Textbox(lines=7, label='Output')
-        
-    gen.click(fn=predict_zh, inputs=[model_input], outputs=outputs)
-    clr.click(fn=lambda value: gr.update(value=""), inputs=clr, outputs=model_input)
-    
+    gr.Markdown("We provide both English and Chinese GENIUS models.")
+    with gr.Tab("Enghlish"):
+        input1 = gr.Textbox(lines=5, placeholder="<mask> Conference on Empirical Methods <mask> submission of research papers <mask> Deep Learning <mask>")
+        output1 = gr.Textbox(lines=5)
+        button1 = gr.Button("Generate")
+    with gr.Tab("Chinese"):
+        input2 = gr.Textbox(lines=5, placeholder="自然语言处理[MASK]谷歌公司[MASK]通用人工智能[MASK]")
+        output2 = gr.Textbox(lines=5)
+        button2 = gr.Button("Generate")
+
+    # with gr.Accordion("Open for More!"):
+    #     gr.Markdown("Look at me...")
+
+    button1.click(predict_en, inputs=input1, outputs=output1)
+    button2.click(predict_zh, inputs=input2, outputs=output2)
+
 demo.launch()