Create README.md

README.md

@@ -0,0 +1,35 @@
+---
+language:
+- vi
+base_model:
+- VietAI/vit5-base
+---
+A model fine-tuned for sentiment analysis based on [VietAI/vit5-base](https://huggingface.co/vinai/phobert-base).
+
+Labels:
+- NEG: Negative
+- POS: Positive
+- NEU: Neutral
+
+Dataset: Comments on Shoppe (https://shopee.vn/)
+## Usage
+```python
+import torch
+from transformers import AutoTokenizer, AutoModelForSequenceClassification
+
+tokenizer = AutoTokenizer.from_pretrained("lamsytan/sentiment-analysis-base-phobert")
+model = AutoModelForSequenceClassification.from_pretrained("lamsytan/sentiment-analysis-product-comment-phobert-base")
+
+sentence = "Áo đẹp lắm nhá lần sau sẽ ghé tiếp ạ"
+
+inputs = tokenizer(sentence, return_tensors="pt", padding=True, truncation=True)
+
+with torch.no_grad():
+    outputs = model(**inputs)
+    logits = outputs.logits
+    probabilities = torch.softmax(logits, dim=-1)
+
+print(probabilities.tolist())
+    # Output:
+    # [[0.010827462188899517, 0.9538241624832153, 0.035348404198884964]]
+    #         ^                         ^                  ^