showme commited on
Commit
ec5a030
·
verified ·
1 Parent(s): 309ab9e

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"] == "clean":
22
+ item["score"] = 1 - item["score"]
23
  return {"result": result}
24
 
25
  if __name__ == "__main__":