Spaces:
Runtime error
Runtime error
Alex Martin
commited on
Commit
•
0ce143e
1
Parent(s):
64a570b
Update app.py
Browse files
app.py
CHANGED
|
@@ -19,7 +19,7 @@ def get_cats(text):
|
|
| 19 |
tokenizer = AutoTokenizer.from_pretrained("aim9061/fine-tuned-toxic-tweet-dilbert")
|
| 20 |
token = tokenizer(text, return_tensors="pt")
|
| 21 |
|
| 22 |
-
outputs = model(**
|
| 23 |
|
| 24 |
prob = torch.sigmoid(output.logits).detach().numpy()[0]
|
| 25 |
|
|
|
|
| 19 |
tokenizer = AutoTokenizer.from_pretrained("aim9061/fine-tuned-toxic-tweet-dilbert")
|
| 20 |
token = tokenizer(text, return_tensors="pt")
|
| 21 |
|
| 22 |
+
outputs = model(**token)
|
| 23 |
|
| 24 |
prob = torch.sigmoid(output.logits).detach().numpy()[0]
|
| 25 |
|