Spaces:
Runtime error
Runtime error
Commit
路
8c2b3e8
1
Parent(s):
ab06164
Update app.py
Browse files
app.py
CHANGED
@@ -16,7 +16,7 @@ def generate(text):
|
|
16 |
outputs = model(inputs['input_ids'], inputs['attention_mask']).detach().numpy()[:3]
|
17 |
# 銈广偝銈€倰绠楀嚭
|
18 |
minimum = np.min(outputs)
|
19 |
-
if
|
20 |
outputs = outputs - minimum
|
21 |
score = outputs / np.sum(outputs)
|
22 |
prediction = f'Not Offensive: {outputs[0]}, Gray-area: {outputs[1]}, Offensive: {outputs[1]}'
|
|
|
16 |
outputs = model(inputs['input_ids'], inputs['attention_mask']).detach().numpy()[:3]
|
17 |
# 銈广偝銈€倰绠楀嚭
|
18 |
minimum = np.min(outputs)
|
19 |
+
if minimum < 0:
|
20 |
outputs = outputs - minimum
|
21 |
score = outputs / np.sum(outputs)
|
22 |
prediction = f'Not Offensive: {outputs[0]}, Gray-area: {outputs[1]}, Offensive: {outputs[1]}'
|