Spaces:
showme
/
Runtime error

showme commited on
Commit
6439801
·
verified ·
1 Parent(s): 4582bba

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