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)
|