samidh commited on
Commit
189c02e
·
verified ·
1 Parent(s): 6b6edbb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -113,9 +113,9 @@ def predict(content, policy):
113
  decoded_output = tokenizer.decode([predicted_token_id])
114
 
115
  if decoded_output == '1':
116
- return f'LABELED (i.e., violating)'
117
  else:
118
- return f'NOT Labeled (i.e., NON-violating)'
119
 
120
 
121
  # Create the interface
 
113
  decoded_output = tokenizer.decode([predicted_token_id])
114
 
115
  if decoded_output == '1':
116
+ return f'True (i.e., VIOLATING)'
117
  else:
118
+ return f'False (i.e., NON-violating)'
119
 
120
 
121
  # Create the interface