showme commited on
Commit
f659a9f
·
verified ·
1 Parent(s): b73645d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
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__":