Update app.py

app.py

@@ -51,8 +51,15 @@ tokenizer = AutoTokenizer.from_pretrained(model_name)
 def refusal_condition(query):
     # 不要再問這些問題啦！
 
-    is_including_tw = '台灣' in query or '台湾' in query or 'taiwan' in query.lower() or 'tw' in query.lower() or '中華民國' in query or '中华民国' in query
-    is_including_cn = '中國' in query or '中国' in query or 'china' in query.lower() or 'cn' in query.lower() or '大陸' in query or '內地' in query or '大陆' in query or '内地' in query or '中華人民共和國' in query or '中华人民共和国' in query
+    query_remove_space = query.replace(' ', '').lower()
+    is_including_tw = False
+    for x in ['台灣', '台湾', 'taiwan', 'tw', '中華民國', '中华民国']:
+        if x in query_remove_space:
+            is_including_tw = True
+    is_including_cn = False
+    for x in ['中國', '中国', 'china', 'cn', '大陸', '內地', '大陆', '内地', '中華人民共和國', '中华人民共和国']:
+        if x in query_remove_space:
+            is_including_cn = True
     if is_including_tw and is_including_cn:
         return True