Spaces:
Runtime error
Runtime error
Alex Martin
commited on
Commit
•
bf1a463
1
Parent(s):
b916553
Update app.py
Browse files
app.py
CHANGED
@@ -22,7 +22,7 @@ def get_cats(text):
|
|
22 |
model = AutoModelForSequenceClassification.from_pretrained("aim9061/fine-tuned-toxic-tweet-dilbert")
|
23 |
outputs = model(**token)
|
24 |
|
25 |
-
prob = torch.sigmoid(
|
26 |
|
27 |
data = create_dict(prob, text)
|
28 |
st.table(data)
|
|
|
22 |
model = AutoModelForSequenceClassification.from_pretrained("aim9061/fine-tuned-toxic-tweet-dilbert")
|
23 |
outputs = model(**token)
|
24 |
|
25 |
+
prob = torch.sigmoid(outputs.logits).detach().numpy()[0]
|
26 |
|
27 |
data = create_dict(prob, text)
|
28 |
st.table(data)
|