update

README.md

@@ -4,7 +4,7 @@ emoji: 🐠
 colorFrom: purple
 colorTo: green
 sdk: gradio
-sdk_version: 3.41.2
+sdk_version: 4.44.1
 app_file: main.py
 pinned: false
 ---

main.py

@@ -156,14 +156,22 @@ def main():
 
             yield output
 
-    model_name_choices = ["trained_models/lip_service_4chan", "trained_models/chinese_porn_novel"] \
-        if platform.system() == "Windows" else \
+    model_name_choices = [
+        "trained_models/lip_service_4chan",
+        "trained_models/chinese_porn_novel"
+    ] if platform.system() == "Windows" else \
         [
             "qgyd2021/lip_service_4chan", "qgyd2021/chinese_chitchat",
-            "qgyd2021/chinese_porn_novel", "qgyd2021/few_shot_intent_gpt2",
+            "qgyd2021/chinese_porn_novel", "qgyd2021/few_shot_intent_gpt2_base",
             "qgyd2021/similar_question_generation",
         ]
 
+    # model_name_choices = [
+    #     "qgyd2021/lip_service_4chan", "qgyd2021/chinese_chitchat",
+    #     "qgyd2021/chinese_porn_novel", "qgyd2021/few_shot_intent_gpt2_base",
+    #     "qgyd2021/similar_question_generation",
+    # ]
+
     demo = gr.Interface(
         fn=fn_stream,
         inputs=[

requirements.txt

@@ -1,9 +1,9 @@
-gradio==3.41.2
-pydantic==1.10.12
+gradio==4.44.1
+pydantic==2.7.4
 thinc==7.4.6
 spacy==2.3.9
 transformers==4.30.2
 numpy==1.21.4
 tqdm==4.62.3
 torch==1.13.0
-datasets
+datasets==2.19.2