update

examples.json

@@ -0,0 +1,10 @@
+[
+    ["怎样擦屁股才能擦的干净", 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.8, "qgyd2021/chinese_porn_novel", false],
+    ["男人走进房间, 上床, 压上", 512, 0.75, 0.35, 1.8, "qgyd2021/chinese_porn_novel", false],
+    [
+        "电销场景意图识别。如果不能确定，请输出 “未知意图”。\n\nExamples:\n------------\ntext: 没关系啦 知道的\nintent: 肯定答复\n------------\ntext: 怎么能联系你\nintent: 查联系方式\n------------\ntext: 恩。让我想想吧。\nintent: 考虑一下\n------------\ntext: 说点有用的\nintent: 请讲重点\n------------\ntext: 唉唉\nintent: 语气词\n------------\ntext: 说快一点\nintent: 请讲重点\n------------\ntext: 再介绍一下\nintent: 要求复述\n------------\ntext: 从哪弄到我信息\nintent: 质疑隐私安全\n------------\ntext: 哎。。不是的\nintent: 不是\n------------\ntext: 给我电话号码\nintent: 查联系方式\n------------\ntext: 先看看吧\nintent: 考虑一下\n------------\ntext: 怎么知道道我的信息\nintent: 质疑隐私安全\n------------\ntext: 哎,再说吧,我再想想\nintent: 考虑一下\n------------\ntext: 不,我清醒。\nintent: 不是\n------------\ntext: 重说一次\nintent: 要求复述\n------------\ntext: 行了,晚安\nintent: 肯定答复\n------------\ntext: 额额额额\nintent: 语气词\n------------\ntext: 恩。哎再说吧我考虑一下hiahia\nintent:\n",
+        128, 0.75, 0.10, 1.2, "qgyd2021/few_shot_intent", true
+    ]
+]

main.py

@@ -2,9 +2,11 @@
 # -*- coding: utf-8 -*-
 import argparse
 from collections import defaultdict
+import json
 import os
 import platform
 import re
+from typing import List
 
 from project_settings import project_path
 
@@ -27,20 +29,15 @@ def get_args():
     parser.add_argument("--repetition_penalty", default=1.0, type=float)
     parser.add_argument('--device', default="cuda" if torch.cuda.is_available() else "cpu", type=str)
 
+    parser.add_argument(
+        "--examples_json_file",
+        default="examples.json",
+        type=str
+    )
     args = parser.parse_args()
     return args
 
 
-description = """
-## GPT2 Chat
-"""
-
-
-examples = [
-
-]
-
-
 def repl1(match):
     result = "{}{}".format(match.group(1), match.group(2))
     return result
@@ -54,6 +51,14 @@ def repl2(match):
 def main():
     args = get_args()
 
+    description = """
+    ## GPT2 Chat
+    """
+
+    # example json
+    with open(args.examples_json_file, "r", encoding="utf-8") as f:
+        examples = json.load(f)
+
     if args.device == 'auto':
         device = 'cuda' if torch.cuda.is_available() else 'cpu'
     else:
@@ -128,7 +133,10 @@ def main():
 
     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"]
+        [
+            "qgyd2021/lib_service_4chan", "qgyd2021/chinese_chitchat",
+            "qgyd2021/chinese_porn_novel", "qgyd2021/few_shot_intent"
+        ]
 
     demo = gr.Interface(
         fn=fn_stream,
@@ -142,12 +150,7 @@ def main():
             gr.Checkbox(value=True, label="is_chat")
         ],
         outputs=[output_text_box],
-        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.8, "qgyd2021/chinese_porn_novel", False],
-            ["男人走进房间, 上床, 压上", 512, 0.75, 0.35, 1.8, "qgyd2021/chinese_porn_novel", False],
-        ],
+        examples=examples,
         cache_examples=False,
         examples_per_page=50,
         title="GPT2 Chat",