mecoaoge2 commited on
Commit
0c0a1dc
1 Parent(s): 4cc3454

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -2
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** Install `transformers` with pip: `pip install transformers` and `SentencePiece` with `pip install SentencePiece`
 
 
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
+ ```