curryaNa commited on
Commit
36b6acb
·
verified ·
1 Parent(s): 6999402

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -30,7 +30,7 @@ def inference(input_text):
30
  predicted_class_id = logits.argmax().item()
31
  output = model.config.id2label[predicted_class_id]
32
  ans = "积极"
33
- if output == 0:
34
  ans = "消极"
35
  return ans
36
 
 
30
  predicted_class_id = logits.argmax().item()
31
  output = model.config.id2label[predicted_class_id]
32
  ans = "积极"
33
+ if output == "0":
34
  ans = "消极"
35
  return ans
36