nashtur
/

postbox_v1

text2text-generation

Model card Files Files and versions

nashtur commited on Dec 1, 2022

Commit

943caa2

·

1 Parent(s): 3f59bc8

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -12,7 +12,7 @@ tags:
 ```
 import torch
 from transformers import PegasusForConditionalGeneration, PegasusTokenizer
-model_name = 'tuner007/pegasus_paraphrase'
 torch_device = 'cuda' if torch.cuda.is_available() else 'cpu'
 tokenizer = PegasusTokenizer.from_pretrained(model_name)
 model = PegasusForConditionalGeneration.from_pretrained(model_name).to(torch_device)

 ```
 import torch
 from transformers import PegasusForConditionalGeneration, PegasusTokenizer
+model_name = 'nashtur/postbox'
 torch_device = 'cuda' if torch.cuda.is_available() else 'cpu'
 tokenizer = PegasusTokenizer.from_pretrained(model_name)
 model = PegasusForConditionalGeneration.from_pretrained(model_name).to(torch_device)