Update app.py
Browse files
app.py
CHANGED
@@ -16,6 +16,10 @@ class TextRequest(BaseModel):
|
|
16 |
@app.post("/predict")
|
17 |
async def predict(request: TextRequest):
|
18 |
result = sentiment_model(request.text)
|
|
|
|
|
|
|
|
|
19 |
return {"result": result}
|
20 |
|
21 |
if __name__ == "__main__":
|
|
|
16 |
@app.post("/predict")
|
17 |
async def predict(request: TextRequest):
|
18 |
result = sentiment_model(request.text)
|
19 |
+
# 遍历 result 列表,检查每个元素的 label
|
20 |
+
for item in result:
|
21 |
+
if item["label"] == "Human":
|
22 |
+
item["score"] = 1 - item["score"]
|
23 |
return {"result": result}
|
24 |
|
25 |
if __name__ == "__main__":
|