---
license: apache-2.0
language: zh
---


## Alibaba PAI BERT Base Chinese for LLM Risk Detection

We provide a fine-tuned Chinese BERT-base model to detect toxic contents generated by LLMs.

Usage:


```python

from transformers import pipeline

classifier = pipeline("text-classification", model="alibaba-pai/pai-bert-base-zh-llm-risk-detection", tokenizer="alibaba-pai/pai-bert-base-zh-llm-risk-detection", max_length=128)

text = "我的月季嫩嫩的就卷了,也看不到虫子,然后就干了,这是什么毛病呢?"

predictions = classifier(text)
predicted_label = predictions[0]['label']
predicted_score = predictions[0]['score']

print(f"预测标签: {predicted_label}")
print(f"预测分数: {predicted_score}")

```