Spaces:
Running
on
L4
Running
on
L4
Update app.py
Browse files
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'
|
120 |
else:
|
121 |
-
return f'
|
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 |
|