Update sents/words examples

ubee/__main__.py

@@ -41,6 +41,7 @@ def greet1(name):
 def greet(
     text1,
     text2,
+    # segment: str
     thresh: float
 ) -> Tuple[pd.DataFrame, pd.DataFrame, pd.DataFrame]:
     """Take inputs, return outputs.
@@ -103,29 +104,55 @@ def main():
 
         Stay tuned for more details coming soon...
         """).strip()
-    examples = [
-        ["yo\nme", "你\n我", .5],
-        ["me\nshe", "你\n她", .5],
+
+    ex1_zh = [
+        '雪开始下大了。',
+        '我握住门柄又试一回。',
+        '这时一个没穿外衣的年轻人，扛着一根草耙，在后面院子里出现了。',        '他招呼我跟着他走，穿过了一个洗衣房和一片铺平的地，那儿有煤棚、抽水机和鸽笼，我们终于到了我上次被接待过的那间温暖的、热闹的大屋子。',
+        '煤、炭和木材混合在一起燃起的熊熊炉火，使这屋子放着光彩。', '在准备摆上丰盛晚餐的桌旁，我很高兴地看到了那位“太太”，以前我从未料想到会有这么一个人存在的。',
+        '我鞠躬等候，以为她会叫我坐下。',
+        '她望望我，往她的椅背一靠，不动，也不出声。'
+    ]
+    ex1_en = [
+        'The snow began to drive thickly.',
+         'I seized the handle to essay another trial; when a young man without coat, and shouldering a pitchfork, appeared in the yard behind.',
+         'He hailed me to follow him, and, after marching through a wash-house, and a paved area containing a coal shed, pump, and pigeon cot, we at length arrived in the huge, warm, cheerful apartment, where I was formerly received.',
+         "It glowed delightfully in the radiance of an immense fire, compounded of coal, peat, and wood; and near the table, laid for a plentiful evening meal, I was pleased to observe the `missis', an individual whose existence I had never previously suspected.",
+         'I bowed and waited, thinking she would bid me take a seat.',
+         'She looked at me, leaning back in her chair, and remained motionless and mute.'
     ]
+    shuffle(ex1_en)
 
+    ex2_zh = 
+    ex2_en = "She looked at me leaning back in her chair and remained motionless and mute".split()
+    shuffle(ex2_en)
+
+    examples = [
+        [ex2_zh, ex2_en, .3],
+        [text_zh, text_en, .5],
+    ]
     lines = 15
     placeholder = "Type or paste text here"
-    default1 = text_zh
-    default2 = text_en
     label1 = "text1"
     label2 = "text2"
     inputs = [
         gr.inputs.Textbox(
-            lines=lines, placeholder=placeholder, default=default1, label=label1
+            lines=lines, placeholder=placeholder, default=ex1_zh, label=label1
         ),
         gr.inputs.Textbox(
-            lines=lines, placeholder=placeholder, default=default2, label=label2
+            lines=lines, placeholder=placeholder, default=ex1_en, label=label2
         ),
+        gr.inputs.Radio(
+            ["para", "sent", "word"],
+            default="para",
+            label="segment"
+        )
         gr.inputs.Slider(
             minimum=0.0,
             maximum=1.0,
             step=0.1,
-            default=0.5,
+            default=0.4,
+            label="threshold",
         ),
     ]
 
@@ -169,7 +196,7 @@ def main():
         article=article,
         # inputs="text",
         # outputs="text",
-        inputs=inputs,
+        inputs=inputs,  # text1, text2, segment, thresh
         outputs=outputs,
         examples=examples,
         # enable_queue=True,

ubee/ubee.py

@@ -31,8 +31,8 @@ def ubee(
     for seq in sents_zh:
         ic(seq)
         label, likelihood = uclas(seq, labels, thresh=thresh)
-        likelihood = round(float(likelihood), 2)
         if label:
+            likelihood = round(float(likelihood), 2)
             res.append((seq, label, likelihood))
             try:
                 lo2.remove(label)