samidh commited on
Commit
25631ed
·
verified ·
1 Parent(s): 2576b91

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -116,9 +116,9 @@ def predict(content, policy):
116
  decoded_output = tokenizer.decode([predicted_token_id])
117
 
118
  if decoded_output == '1':
119
- return f'True (i.e., Meets Label Criteria)'
120
  else:
121
- return f'False (i.e., Does NOT Meet Label Criteria)'
122
 
123
  # Create the interface
124
 
 
116
  decoded_output = tokenizer.decode([predicted_token_id])
117
 
118
  if decoded_output == '1':
119
+ return f'1: One or more policy labels apply'
120
  else:
121
+ return f'0: None of the policy labels apply'
122
 
123
  # Create the interface
124