Update app.py

app.py

@@ -16,15 +16,22 @@ class TextRequest(BaseModel):
 @app.post("/predict")
 async def predict(request: TextRequest):
     result = sentiment_model(request.text)
-    # 遍历 result 列表，检查每个元素的 label
-    for item in result:
-        if item["label"] == "LABEL_0":
-            item["score"] = 1 - item["score"]
-        elif item["label"] == "LABEL_1":
-            item["label"] = "AI"
-    print("处理后的 result:", result)
-    return {"result": result}
-
+    print("处理前的 result:", result)
+    
+    # 获取原始预测结果
+    original = result[0]
+    
+    # 计算AI的概率
+    ai_probability = original["score"] if original["label"] == "LABEL_1" else 1 - original["score"]
+    
+    processed_result = [{
+        "label": "AI" if ai_probability > 0.5 else "Human",
+        "score": ai_probability
+    }]
+    
+    print("处理后的 result:", processed_result)
+    return {"result": processed_result}
+    
 if __name__ == "__main__":
     import uvicorn
     uvicorn.run(app, host="0.0.0.0", port=7860)