VietAI
/

vit5-large-vietnews-summarization

text2text-generation

Inference Endpoints

text-generation-inference

Model card Files Files and versions Community

razent commited on May 12, 2022

Commit

ce84304

•

1 Parent(s): 5b80a1c

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -23,7 +23,7 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 tokenizer = AutoTokenizer.from_pretrained("VietAI/vit5-large-vietnews-summarization")
 model = AutoModelForSeq2SeqLM.from_pretrained("VietAI/vit5-large-vietnews-summarization")
-sentence = "Xin chào"
 text =  "summarize: " + sentence + " </s>"
 encoding = tokenizer.encode_plus(text, pad_to_max_length=True, return_tensors="pt")
 input_ids, attention_masks = encoding["input_ids"].to("cuda"), encoding["attention_mask"].to("cuda")

 tokenizer = AutoTokenizer.from_pretrained("VietAI/vit5-large-vietnews-summarization")
 model = AutoModelForSeq2SeqLM.from_pretrained("VietAI/vit5-large-vietnews-summarization")
+sentence = "VietAI là tổ chức phi lợi nhuận với sứ mệnh ươm mầm tài năng về trí tuệ nhân tạo và xây dựng một cộng đồng các chuyên gia trong lĩnh vực trí tuệ nhân tạo đẳng cấp quốc tế tại Việt Nam."
 text =  "summarize: " + sentence + " </s>"
 encoding = tokenizer.encode_plus(text, pad_to_max_length=True, return_tensors="pt")
 input_ids, attention_masks = encoding["input_ids"].to("cuda"), encoding["attention_mask"].to("cuda")