Upload app.py

app.py

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