addy88 commited on
Commit
6ecb352
1 Parent(s): dc8652b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -2,8 +2,8 @@
2
  Here is how to use this model in PyTorch:
3
  ```python
4
  from transformers import PerceiverTokenizer, PerceiverForMaskedLM
5
- tokenizer = PerceiverTokenizer.from_pretrained("deepmind/language-perceiver")
6
- model = PerceiverForMaskedLM.from_pretrained("deepmind/language-perceiver")
7
  text = "This is an incomplete sentence where some words are missing."
8
  # prepare input
9
  encoding = tokenizer(text, padding="max_length", return_tensors="pt")
 
2
  Here is how to use this model in PyTorch:
3
  ```python
4
  from transformers import PerceiverTokenizer, PerceiverForMaskedLM
5
+ tokenizer = PerceiverTokenizer.from_pretrained("addy88/perceiver_imdb")
6
+ model = PerceiverForMaskedLM.from_pretrained("addy88/perceiver_imdb")
7
  text = "This is an incomplete sentence where some words are missing."
8
  # prepare input
9
  encoding = tokenizer(text, padding="max_length", return_tensors="pt")