Upload app.py

app.py

@@ -1109,10 +1109,10 @@ def netOrNotnet(
     user_input, chatbot, max_length, top_p, temperature, history, past_key_values
 ):
     user_input_original = user_input
-    user_input_prompt = '请判断这个指令:“' + user_input + '”是否需要通过网络搜索才能准确完成，你只需要回答"是"或者"否",不要其它多余的内容，更不要回答"不需要"或者“需要”'
+    user_input_prompt = '请判断这个指令:“' + user_input + '”是否需要通过网络搜索才能准确完成，你只需要回答"是"或者"否",不要其它多余的内容'
     
     response, history = model.chat(tokenizer, user_input_prompt,top_p=0.2, temperature=0.85, history=[])
-    ytkeywords = ['是。', '不清楚', '无法', '是否', '不知道', '需要通过网络', '取决于']
+    ytkeywords = ['否。', '不需要通过网络']
 
     NET = any(keyword in response for keyword in ytkeywords)
 
@@ -1120,7 +1120,8 @@ def netOrNotnet(
     print(response)
     print(NET)
     if NET:
-        yield from GGSearch(
+        yield from predict(
+            RETRY_FLAG,# type: ignore
             user_input_original,
             chatbot,
             max_length,
@@ -1131,9 +1132,9 @@ def netOrNotnet(
         )
 
 
+
     else:
-        yield from predict(
-            RETRY_FLAG,# type: ignore
+        yield from GGSearch(
             user_input_original,
             chatbot,
             max_length,