Create app.py

app.py

@@ -0,0 +1,18 @@
+from fastapi import FastAPI
+from transformers import pipeline
+
+app = FastAPI()
+
+# 加载预训练的文本分类模型（建议替换为你微调后的 BERT 模型路径）
+# 例如：'shibing624/bert4confusing-base-chinese' 或自定义模型
+classifier = pipeline("text-classification", model="bert-base-chinese")
+
+@app.get("/")
+def home():
+    return {"message": "Text Moderation API is running."}
+
+@app.post("/analyze")
+def analyze_text(text: str):
+    # 执行推理
+    result = classifier(text)
+    return {"result": result}