update

main.py

@@ -76,7 +76,7 @@ def main():
     args = get_args()
 
     description = """
-    ## GPT2 Chat
+    ## Generate Similar Question
     """
 
     # example json
@@ -96,7 +96,7 @@ def main():
                   top_p: float = 0.85,
                   temperature: float = 0.35,
                   repetition_penalty: float = 1.2,
-                  model_name: str = "qgyd2021/lip_service_4chan",
+                  model_name: str = "qgyd2021/similar_question_generation",
                   is_chat: bool = True,
                   ):
         tokenizer = BertTokenizer.from_pretrained(model_name)
@@ -156,11 +156,9 @@ def main():
 
             yield output
 
-    model_name_choices = ["trained_models/lip_service_4chan", "trained_models/chinese_porn_novel"] \
+    model_name_choices = ["trained_models/similar_question_generation",] \
         if platform.system() == "Windows" else \
         [
-            "qgyd2021/lip_service_4chan", "qgyd2021/chinese_chitchat",
-            "qgyd2021/chinese_porn_novel", "qgyd2021/few_shot_intent",
             "qgyd2021/similar_question_generation"
         ]
 
@@ -179,7 +177,7 @@ def main():
         examples=examples,
         cache_examples=False,
         examples_per_page=50,
-        title="GPT2 Chat",
+        title="Generate Similar Question",
         description=description,
     )
     demo.queue().launch()