Update README.md
Browse files
README.md
CHANGED
|
@@ -35,7 +35,7 @@ You'll need to pip install transformers & maybe sentencepiece
|
|
| 35 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
| 36 |
import torch, time
|
| 37 |
device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
|
| 38 |
-
model_name = '/
|
| 39 |
max_seq_len = 510
|
| 40 |
|
| 41 |
tokenizer = AutoTokenizer.from_pretrained(model_name, model_max_length=max_seq_len)
|
|
|
|
| 35 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
| 36 |
import torch, time
|
| 37 |
device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
|
| 38 |
+
model_name = 'Reggie/muppet-roberta-base-joke_detector'
|
| 39 |
max_seq_len = 510
|
| 40 |
|
| 41 |
tokenizer = AutoTokenizer.from_pretrained(model_name, model_max_length=max_seq_len)
|