Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -21,23 +21,50 @@ labels = ["Non-manipulative / 非操纵性", "Manipulative / 操纵性"]
|
|
| 21 |
def classify(text):
|
| 22 |
inputs = tokenizer(text, return_tensors="pt", truncation=True, padding=True).to(device)
|
| 23 |
with torch.no_grad():
|
| 24 |
-
|
| 25 |
-
|
| 26 |
-
|
| 27 |
-
|
| 28 |
-
|
| 29 |
-
|
| 30 |
-
|
| 31 |
-
|
| 32 |
|
| 33 |
|
| 34 |
# Gradio 界面
|
| 35 |
interface = gr.Interface(
|
| 36 |
fn=classify,
|
| 37 |
-
inputs=gr.Textbox(
|
| 38 |
-
|
| 39 |
-
|
| 40 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 41 |
)
|
| 42 |
|
|
|
|
| 43 |
interface.launch()
|
|
|
|
| 21 |
def classify(text):
|
| 22 |
inputs = tokenizer(text, return_tensors="pt", truncation=True, padding=True).to(device)
|
| 23 |
with torch.no_grad():
|
| 24 |
+
outputs = model(**inputs)
|
| 25 |
+
probs = torch.softmax(outputs.logits, dim=1)[0] # 取第一个样本的概率向量
|
| 26 |
+
probs = torch.clamp(probs, max=0.95) # 限制最大置信度为 95%
|
| 27 |
+
result = "🧠 预测 / Prediction:\n"
|
| 28 |
+
for i, label in enumerate(labels):
|
| 29 |
+
percent = round(probs[i].item() * 100, 2)
|
| 30 |
+
result += f"{label}: {percent}%\n"
|
| 31 |
+
return result
|
| 32 |
|
| 33 |
|
| 34 |
# Gradio 界面
|
| 35 |
interface = gr.Interface(
|
| 36 |
fn=classify,
|
| 37 |
+
inputs=gr.Textbox(
|
| 38 |
+
lines=4,
|
| 39 |
+
placeholder="Enter a sentence in English or Chinese... / 输入英文或中文句子",
|
| 40 |
+
label="📝 Input Text / 输入文本"
|
| 41 |
+
),
|
| 42 |
+
outputs=gr.Textbox(label="📊 Prediction / 预测结果"),
|
| 43 |
+
title="🧠 Manipulative Language Detector / 操纵性语言识别器",
|
| 44 |
+
description="""
|
| 45 |
+
🔍 This tool detects **emotionally manipulative language** in English or Chinese digital communication.
|
| 46 |
+
🧪 The model was fine-tuned on a manually annotated dataset of 10,000 Chinese messages, categorized into four manipulation types.
|
| 47 |
+
|
| 48 |
+
---
|
| 49 |
+
|
| 50 |
+
📌 **Disclaimer / 免责声明:**
|
| 51 |
+
This system is for **research and educational purposes only**.
|
| 52 |
+
It **does not guarantee accuracy** and **should not be used as legal or clinical evidence**.
|
| 53 |
+
|
| 54 |
+
本工具仅用于**学术研究与教学演示**,不构成法律、医疗或其他正式用途的依据。
|
| 55 |
+
|
| 56 |
+
---
|
| 57 |
+
|
| 58 |
+
🤖 **Model Info**:
|
| 59 |
+
- Model: `LilithHu/mbert-manipulative-detector`
|
| 60 |
+
- Base: `mDeBERTa-v3` multilingual pre-trained model
|
| 61 |
+
- Fine-tuned using HuggingFace Transformers on labeled Chinese data
|
| 62 |
+
|
| 63 |
+
🌐 Built with Gradio and hosted on HuggingFace Spaces.
|
| 64 |
+
""",
|
| 65 |
+
theme="default",
|
| 66 |
+
allow_flagging="never"
|
| 67 |
)
|
| 68 |
|
| 69 |
+
|
| 70 |
interface.launch()
|