Forturne
/

KPMG-NARVIS-summarization

text2text-generation

Model card Files Files and versions

Forturne commited on Feb 13, 2023

Commit

f323473

·

1 Parent(s): 7f381cb

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -5,8 +5,8 @@ import torch
 from transformers import PreTrainedTokenizerFast
 from transformers import BartForConditionalGeneration
-tokenizer = PreTrainedTokenizerFast.from_pretrained('KPMG-NARVIS-summarization')
-model = BartForConditionalGeneration.from_pretrained('KPMG-NARVIS-summarization')
 device="cuda:0"
 model.to(device)

 from transformers import PreTrainedTokenizerFast
 from transformers import BartForConditionalGeneration
+tokenizer = PreTrainedTokenizerFast.from_pretrained('Forturne/KPMG-NARVIS-summarization')
+model = BartForConditionalGeneration.from_pretrained('Forturne/KPMG-NARVIS-summarization')
 device="cuda:0"
 model.to(device)