samidh commited on
Commit
a58db7b
·
verified ·
1 Parent(s): 3e9b843

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
 
124
  # Create the interface
 
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
 
124
  # Create the interface