test

app.py

@@ -95,16 +95,39 @@ block = gr.Blocks(css=css)
 
 examples = [
     [
-        '戴着眼镜的猫',
-        '油画'
+        'text_summarization',
+        '在芬兰、瑞典提交“入约”申请近一个月来，北约成员国内部尚未对此达成一致意见。与此同时，俄罗斯方面也多次对北约“第六轮扩张”发出警告。据北约官网显示，北约秘书长斯托尔滕贝格将于本月12日至13日出访瑞典和芬兰，并将分别与两国领导人进行会晤。',
+        4, 512, 0.0, 1.0
     ],
     [
-        '日落时的城市天际线,史前遗迹风格',
-        '油画'
+        'copywriting_generation',
+        '芍药香氛的沐浴乳',
+        32, 512, 0.9, 1.2
     ],
     [
-        '一只猫坐在椅子上，戴着一副墨镜, low poly 风格',
-        '卡通'
+        'novel_continuation',
+        '昆仑山可以说是天下龙脉的根源，所有的山脉都可以看作是昆仑的分支。这些分出来的枝枝杈杈，都可以看作是一条条独立的龙脉。',
+        2, 512, 0.9, 1.2
+    ],
+    [
+        'answer_generation',
+        '杜鹃花怎么养？',
+        2, 512, 0.9, 1.2
+    ],
+    [
+        'couplet_continuation',
+        '天增岁月人增寿',
+        2, 512, 0.9, 1.0
+    ],
+    [
+        'composition_generation',
+        '诚以养德，信以修身',
+        128, 512, 0.9, 1.2
+    ],
+    [
+        'text_cloze',
+        '她有着一双[MASK]的眼眸。',
+        1, 512, 0.9, 1.0
     ],
 ]
 
@@ -147,8 +170,7 @@ with block:
         with gr.Box():
             with gr.Row().style(mobile_collapse=False, equal_height=True):
                 text = gr.Textbox(
-                    value='戴着眼镜的猫',
-                    label="Prompt (Chinese)",
+                    label="input_text",
                     show_label=False,
                     max_lines=1,
                     placeholder="Enter your Chinese prompt",
@@ -173,21 +195,18 @@ with block:
                            ],
                            value='text_summarization')
 
-        min_dec_len = gr.Number(label="min_dec_len", value=1, interactive=True)
-        seq_len = gr.Number(label="seq_len", value=128, nteractive=True)
-        topp = gr.Number(label="topp", value=1.0, interactive=True, precision=1)
-        penalty_score = gr.Number(label="penalty_score", value=1.0, interactive=True, precision=1)
+        min_dec_len = gr.Slider(minimum=1, maximum=511, value=1, label="min_dec_len", step=1, interactive=True)
+        seq_len = gr.Slider(minimum=2, maximum=512, value=128, label="seq_len", step=1, interactive=True)
+        topp = gr.Slider(minimum=0.0, maximum=1.0, value=1.0, label="topp", step=0.01, interactive=True)
+        penalty_score = gr.Slider(minimum=1.0, maximum=2.0, value=1.0, label="topp", step=0.01, interactive=True)
 
-        text_gen = gr.Text(
-            value='哈哈哈哈哈', label="Generated images", show_label=False
-        )
+        text_gen = gr.Text(label="generated_text")
 
         ex = gr.Examples(examples=examples, fn=inference, inputs=[
-                         text, task], outputs=text_gen, cache_examples=False)
-        ex.dataset.headers = [""]
+                         task, text, min_dec_len, seq_len, topp, penalty_score], outputs=text_gen, cache_examples=False)
 
-        text.submit(inference, inputs=[text, task], outputs=text_gen)
-        btn.click(inference, inputs=[text, task], outputs=text_gen)
+        text.submit(inference, inputs=[task, text, min_dec_len, seq_len, topp, penalty_score], outputs=text_gen)
+        btn.click(inference, inputs=[task, text, min_dec_len, seq_len, topp, penalty_score], outputs=text_gen)
         gr.HTML(
             """
                 <div class="footer">