[update]add model

main.py

@@ -41,11 +41,16 @@ examples = [
 ]
 
 
-def repl(match):
+def repl1(match):
     result = "{}{}".format(match.group(1), match.group(2))
     return result
 
 
+def repl2(match):
+    result = "{}".format(match.group(1))
+    return result
+
+
 def main():
     args = get_args()
 
@@ -105,18 +110,23 @@ def main():
 
             output_ = output_.replace("[UNK] ", "")
             output_ = output_.replace("[UNK]", "")
+            output_ = output_.replace("[CLS] ", "")
+            output_ = output_.replace("[CLS]", "")
 
             output += output_
+            if output.startswith("[SEP]"):
+                output = output[5:]
+
+            output = output.lstrip(" ,.!?")
+            output = re.sub(r"([，。！？\u4e00-\u9fa5]) ([，。！？\u4e00-\u9fa5])", repl1, output)
+            output = re.sub(r"([，。！？\u4e00-\u9fa5]) ", repl2, output)
 
-            output = output.lstrip("[SEP] ,.!?")
+            output = output.replace("[SEP] ", "\n")
             output = output.replace("[SEP]", "\n")
-            output = re.sub(r"([，。！？\u4e00-\u9fa5]) ([，。！？\u4e00-\u9fa5])", repl, output)
-            output = output.strip()
 
-            output_text_box.value += output
             yield output
 
-    model_name_choices = ["trained_models/lib_service_4chan"] \
+    model_name_choices = ["trained_models/lib_service_4chan", "trained_models/chinese_porn_novel"] \
         if platform.system() == "Windows" else \
         ["qgyd2021/lib_service_4chan", "qgyd2021/chinese_chitchat", "qgyd2021/chinese_porn_novel"]
 
@@ -135,8 +145,8 @@ def main():
         examples=[
             ["怎样擦屁股才能擦的干净", 512, 0.75, 0.35, 1.2, "qgyd2021/lib_service_4chan", True],
             ["你好", 512, 0.75, 0.35, 1.2, "qgyd2021/chinese_chitchat", True],
-            ["白洁走到床边并脱去内衣, 一双硕大的", 512, 0.75, 0.35, 1.2, "qgyd2021/chinese_porn_novel", False],
-            ["男人走进房间, 上床, 压上", 512, 0.75, 0.35, 1.2, "qgyd2021/chinese_porn_novel", False],
+            ["白洁走到床边并脱去内衣, 一双硕大的", 512, 0.75, 0.35, 1.8, "qgyd2021/chinese_porn_novel", False],
+            ["男人走进房间, 上床, 压上", 512, 0.75, 0.35, 1.8, "qgyd2021/chinese_porn_novel", False],
         ],
         cache_examples=False,
         examples_per_page=50,