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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -116,10 +116,9 @@ def predict(content, policy):
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
125
 
 
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