update

examples.json

@@ -1,5 +1,5 @@
 [
-    ["怎样擦屁股才能擦的干净", 512, 0.75, 0.35, 1.2, "qgyd2021/lib_service_4chan", true],
+    ["怎样擦屁股才能擦的干净", 512, 0.75, 0.35, 1.2, "qgyd2021/lip_service_4chan", true],
     ["你好", 512, 0.75, 0.35, 1.2, "qgyd2021/chinese_chitchat", true],
     ["白洁走到床边并脱去内衣, 一双硕大的", 512, 0.75, 0.35, 1.8, "qgyd2021/chinese_porn_novel", false],
     ["男人走进房间, 上床, 压上", 512, 0.75, 0.35, 1.8, "qgyd2021/chinese_porn_novel", false],

main.py

@@ -96,7 +96,7 @@ def main():
                   top_p: float = 0.85,
                   temperature: float = 0.35,
                   repetition_penalty: float = 1.2,
-                  model_name: str = "qgyd2021/lib_service_4chan",
+                  model_name: str = "qgyd2021/lip_service_4chan",
                   is_chat: bool = True,
                   ):
         tokenizer = BertTokenizer.from_pretrained(model_name)
@@ -156,10 +156,10 @@ def main():
 
             yield output
 
-    model_name_choices = ["trained_models/lib_service_4chan", "trained_models/chinese_porn_novel"] \
+    model_name_choices = ["trained_models/lip_service_4chan", "trained_models/chinese_porn_novel"] \
         if platform.system() == "Windows" else \
         [
-            "qgyd2021/lib_service_4chan", "qgyd2021/chinese_chitchat",
+            "qgyd2021/lip_service_4chan", "qgyd2021/chinese_chitchat",
             "qgyd2021/chinese_porn_novel", "qgyd2021/few_shot_intent"
         ]