Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -51,8 +51,15 @@ tokenizer = AutoTokenizer.from_pretrained(model_name)
|
|
51 |
def refusal_condition(query):
|
52 |
# 不要再問這些問題啦!
|
53 |
|
54 |
-
|
55 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
56 |
if is_including_tw and is_including_cn:
|
57 |
return True
|
58 |
|
|
|
51 |
def refusal_condition(query):
|
52 |
# 不要再問這些問題啦!
|
53 |
|
54 |
+
query_remove_space = query.replace(' ', '').lower()
|
55 |
+
is_including_tw = False
|
56 |
+
for x in ['台灣', '台湾', 'taiwan', 'tw', '中華民國', '中华民国']:
|
57 |
+
if x in query_remove_space:
|
58 |
+
is_including_tw = True
|
59 |
+
is_including_cn = False
|
60 |
+
for x in ['中國', '中国', 'china', 'cn', '大陸', '內地', '大陆', '内地', '中華人民共和國', '中华人民共和国']:
|
61 |
+
if x in query_remove_space:
|
62 |
+
is_including_cn = True
|
63 |
if is_including_tw and is_including_cn:
|
64 |
return True
|
65 |
|