Commit
·
ff594dd
1
Parent(s):
5bd1290
Update README.md
Browse files
README.md
CHANGED
|
@@ -25,7 +25,7 @@ You can use this model directly with a pipeline for masked language modeling:
|
|
| 25 |
|
| 26 |
```python
|
| 27 |
from transformers import pipeline
|
| 28 |
-
model = pipeline(task="fill-mask", model="
|
| 29 |
model("Improved <mask> for playing a game of thumb wrestling.")
|
| 30 |
```
|
| 31 |
|
|
@@ -62,8 +62,8 @@ from transformers import AutoTokenizer, AutoModelForMaskedLM
|
|
| 62 |
# cuda/cpu
|
| 63 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
| 64 |
|
| 65 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
| 66 |
-
model = AutoModelForMaskedLM.from_pretrained("
|
| 67 |
|
| 68 |
TEXT = "Improved <mask> for playing a game of thumb wrestling."
|
| 69 |
|
|
|
|
| 25 |
|
| 26 |
```python
|
| 27 |
from transformers import pipeline
|
| 28 |
+
model = pipeline(task="fill-mask", model="hupd/hupd-distilroberta-base")
|
| 29 |
model("Improved <mask> for playing a game of thumb wrestling.")
|
| 30 |
```
|
| 31 |
|
|
|
|
| 62 |
# cuda/cpu
|
| 63 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
| 64 |
|
| 65 |
+
tokenizer = AutoTokenizer.from_pretrained("hupd/hupd-distilroberta-base")
|
| 66 |
+
model = AutoModelForMaskedLM.from_pretrained("hupd/hupd-distilroberta-base").to(device)
|
| 67 |
|
| 68 |
TEXT = "Improved <mask> for playing a game of thumb wrestling."
|
| 69 |
|