Update app.py

app.py

@@ -16,7 +16,6 @@ polite_rewrite = pipeline(
 
 
 # ---------- 1. 加载 Hugging Face 模型 ----------
-
 # 中 → 英 翻译
 translator = pipeline("translation", model="Helsinki-NLP/opus-mt-zh-en")
 
@@ -30,6 +29,14 @@ formality_cls = pipeline("text-classification", model="LenDigLearn/formality-cla
 hedge_cls = pipeline("text-classification", model="ChrisLiewJY/BERTweet-Hedge")
 
 
+# ---------- 2. 一些简单的中文 & 英文规则打分函数 ----------
+
+POLITE_WORDS_ZH = ["请", "麻烦您", "劳烦", "敬请", "拜托", "打扰了", "烦请"]
+HEDGE_WORDS_ZH = ["是否", "可能", "大概", "也许", "好像", "觉得", "有点"]
+IMPERATIVE_WORDS_ZH = ["必须", "务必", "不得", "不准", "立即", "马上", "必须要"]
+
+
+
 # ---------- 2. 一些简单的中文 & 英文规则打分函数 ----------
 
 POLITE_WORDS_ZH = ["请", "麻烦您", "劳烦", "敬请", "拜托", "打扰了", "烦请"]