Update README.md
Browse files
README.md
CHANGED
@@ -25,9 +25,12 @@ The general architecture and experimental results of ViSoBERT can be found in ou
|
|
25 |
|
26 |
**Please CITE** our paper when ViSoBERT is used to help produce published results or is incorporated into other software.
|
27 |
|
28 |
-
**Installation**
|
|
|
|
|
29 |
|
30 |
**Example usage**
|
|
|
31 |
from transformers import AutoModel,AutoTokenizer
|
32 |
import torch
|
33 |
|
@@ -37,4 +40,5 @@ tokenizer = AutoTokenizer.from_pretrained('uitnlp/visobert')
|
|
37 |
encoding = tokenizer('dau xanh rau ma',return_tensors='pt')
|
38 |
|
39 |
with torch.no_grad():
|
40 |
-
output = model(**encoding)
|
|
|
|
25 |
|
26 |
**Please CITE** our paper when ViSoBERT is used to help produce published results or is incorporated into other software.
|
27 |
|
28 |
+
**Installation**
|
29 |
+
|
30 |
+
Install `transformers` with pip: `pip install transformers` and `SentencePiece` with pip: `pip install SentencePiece`
|
31 |
|
32 |
**Example usage**
|
33 |
+
```python
|
34 |
from transformers import AutoModel,AutoTokenizer
|
35 |
import torch
|
36 |
|
|
|
40 |
encoding = tokenizer('dau xanh rau ma',return_tensors='pt')
|
41 |
|
42 |
with torch.no_grad():
|
43 |
+
output = model(**encoding)
|
44 |
+
```
|