Update README.md

README.md

@@ -42,12 +42,21 @@ The dataset is about 30k Vietnamese news dataset from website thanhnien.vn
 - P/s: I've already tried to deploy my model on Streamlit's cloud, but It was always being broken due to out of memory
 
 
-## Usage (Huggingface)
+## Example usage
 ```
 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+
+
+"""
+Category includes: ['thời sự ', 'thế giới', 'tài chính kinh doanh', 'đời sống', 'văn hoá', 'giải trí', 'giới trẻ', 'giáo dục','công nghệ', 'sức khoẻ']
+"""
+
+category = "thời sự"
+headline = "Nam thanh niên"  # A full headline or only some text
+
 text = f"<|startoftext|> {category} <|headline|> {headline}"
 
 tokenizer = AutoTokenizer.from_pretrained("tuanle/VN-News-GPT2")