Spaces:
Running
on
L4
Running
on
L4
Update app.py
Browse files
app.py
CHANGED
@@ -95,7 +95,7 @@ def predict(content, policy):
|
|
95 |
decoded_output = tokenizer.decode(outputs[0], skip_special_tokens=True)
|
96 |
if decoded_output[-1] == '0':
|
97 |
return 'NON-Violating'
|
98 |
-
else
|
99 |
return 'VIOLATING'
|
100 |
|
101 |
# Create Gradio interface
|
|
|
95 |
decoded_output = tokenizer.decode(outputs[0], skip_special_tokens=True)
|
96 |
if decoded_output[-1] == '0':
|
97 |
return 'NON-Violating'
|
98 |
+
else:
|
99 |
return 'VIOLATING'
|
100 |
|
101 |
# Create Gradio interface
|