Update app.py

app.py

@@ -2,9 +2,10 @@ import gradio as gr
 from transformers import pipeline
 
 pipeline_en = pipeline(task="text2text-generation", model="beyond/genius-large")
-
+# pipeline_en = pipeline_zh 
 pipeline_zh = pipeline(task="text2text-generation", model="beyond/genius-base-chinese")
 
+
 def predict_en(sketch):
   generated_text = pipeline_en(sketch, num_beams=3, do_sample=True, max_length=200)[0]['generated_text']
   return generated_text
@@ -17,14 +18,22 @@ def predict_zh(sketch):
 
 
 with gr.Blocks() as demo:
-    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>")
+    gr.Markdown("""
+                # 💡GENIUS – generating text using sketches! 
+                [Please check our GitHub repo for more details.](https://github.com/beyondguo/genius)
+                
+                We provide both English and Chinese GENIUS models.
+                - For English version, the mask token is `<mask>`;
+                - For Chinese version, the mask token is `[MASK]`.
+                """)
+    with gr.Tab("English"):
+        input1 = gr.Textbox(lines=5, value="<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]")
+        input2 = gr.Textbox(lines=5, value="自然语言处理[MASK]谷歌公司[MASK]通用人工智能[MASK]")
         output2 = gr.Textbox(lines=5)
+        output2 = output2.replace(' ','')
         button2 = gr.Button("Generate")
 
     # with gr.Accordion("Open for More!"):