Reggie commited on
Commit
753c6a4
·
1 Parent(s): a3acc72

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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 = '/path/to/model'
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)