YC-Chen commited on
Commit
cf55a3e
1 Parent(s): e6cde65

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -51,8 +51,8 @@ tokenizer = AutoTokenizer.from_pretrained(model_name)
51
  def refusal_condition(query):
52
  # 不要再問這些問題啦!
53
 
54
- is_including_tw = '台灣' in query or '台湾' in query or 'taiwan' in query.lower() or 'tw' in query.lower()
55
- is_including_cn = '中國' in query or '中国' in query or 'china' in query.lower() or 'cn' in query.lower() or '大陸' in query or '內地' in query
56
  if is_including_tw and is_including_cn:
57
  return True
58
 
 
51
  def refusal_condition(query):
52
  # 不要再問這些問題啦!
53
 
54
+ is_including_tw = '台灣' in query or '台湾' in query or 'taiwan' in query.lower() or 'tw' in query.lower() or '中華民國' in query or '中华民国' in query
55
+ 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
56
  if is_including_tw and is_including_cn:
57
  return True
58