omidvaramin
/

Ht5-small

Text2Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

omidvaramin commited on Sep 26, 2023

Commit

4d9580c

·

1 Parent(s): 6a65c56

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -70,8 +70,8 @@ print(device)
 #appending the task identifier to the beginning of input
 article = "Headline: " + article
-model = AutoModelForSeq2SeqLM.from_pretrained("omidvaramin/HBART").to(device)
-tokenizer = AutoTokenizer.from_pretrained("omidvaramin/HBART")
 #encodign article using tokenizer
 encoding = tokenizer(article
                      , max_length=1024

 #appending the task identifier to the beginning of input
 article = "Headline: " + article
+model = AutoModelForSeq2SeqLM.from_pretrained("omidvaramin/Ht5-small").to(device)
+tokenizer = AutoTokenizer.from_pretrained("omidvaramin/Ht5-small")
 #encodign article using tokenizer
 encoding = tokenizer(article
                      , max_length=1024