Update README.md
Browse files
README.md
CHANGED
@@ -117,11 +117,14 @@ The authors provide additionally notes about the training procedure used, in the
|
|
117 |
```
|
118 |
from transformers import TransfoXLTokenizer, TransfoXLModel
|
119 |
import torch
|
|
|
120 |
tokenizer = TransfoXLTokenizer.from_pretrained("zpbrent/newInfect")
|
121 |
model = TransfoXLModel.from_pretrained("zpbrent/newInfect")
|
|
|
122 |
inputs = tokenizer("Hello, my dog is cute", return_tensors="pt")
|
123 |
outputs = model(**inputs)
|
124 |
-
|
|
|
125 |
```
|
126 |
|
127 |
|
|
|
117 |
```
|
118 |
from transformers import TransfoXLTokenizer, TransfoXLModel
|
119 |
import torch
|
120 |
+
|
121 |
tokenizer = TransfoXLTokenizer.from_pretrained("zpbrent/newInfect")
|
122 |
model = TransfoXLModel.from_pretrained("zpbrent/newInfect")
|
123 |
+
|
124 |
inputs = tokenizer("Hello, my dog is cute", return_tensors="pt")
|
125 |
outputs = model(**inputs)
|
126 |
+
|
127 |
+
print(outputs)
|
128 |
```
|
129 |
|
130 |
|