Update README.md

README.md

@@ -1,3 +1,9 @@
+---
+language:
+- vi
+base_model:
+- vinai/phobert-base
+---
 A model fine-tuned for sentiment analysis based on [vinai/phobert-base](https://huggingface.co/vinai/phobert-base).
 
 Labels:
@@ -7,8 +13,9 @@ Labels:
 
 Dataset: Comments on Shoppe (https://shopee.vn/)
 ## Usage
-from transformers import AutoTokenizer, AutoModelForSequenceClassification
+```python
 import torch
+from transformers import AutoTokenizer, AutoModelForSequenceClassification
 
 tokenizer = AutoTokenizer.from_pretrained("lamsytan/sentiment-analysis-base-phobert")
 model = AutoModelForSequenceClassification.from_pretrained("lamsytan/sentiment-analysis-base-phobert")
@@ -26,5 +33,4 @@ print(probabilities.tolist())
     # Output:
     # [[0.010827462188899517, 0.9538241624832153, 0.035348404198884964]]
     #     ^      ^      ^
-    #    NEG    POS    NEU
-
+    #    NEG    POS    NEU