addy88
/

perceiver_imdb

Text Classification

Model card Files Files and versions

addy88 commited on Jan 2, 2022

Commit

6ecb352

·

1 Parent(s): dc8652b

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -2,8 +2,8 @@
 Here is how to use this model in PyTorch:
 ```python
 from transformers import PerceiverTokenizer, PerceiverForMaskedLM
-tokenizer = PerceiverTokenizer.from_pretrained("deepmind/language-perceiver")
-model = PerceiverForMaskedLM.from_pretrained("deepmind/language-perceiver")
 text = "This is an incomplete sentence where some words are missing."
 # prepare input
 encoding = tokenizer(text, padding="max_length", return_tensors="pt")

 Here is how to use this model in PyTorch:
 ```python
 from transformers import PerceiverTokenizer, PerceiverForMaskedLM
+tokenizer = PerceiverTokenizer.from_pretrained("addy88/perceiver_imdb")
+model = PerceiverForMaskedLM.from_pretrained("addy88/perceiver_imdb")
 text = "This is an incomplete sentence where some words are missing."
 # prepare input
 encoding = tokenizer(text, padding="max_length", return_tensors="pt")