imvladikon
/

het5_summarization

text2text-generation

text-generation-inference

Model card Files Files and versions

imvladikon commited on Jul 6, 2023

Commit

013e7d3

·

1 Parent(s): 51b722d

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -9,7 +9,7 @@ pipeline_tag: summarization
 from transformers import AutoModelForSeq2SeqLM, AutoTokenizer, SummarizationPipeline
-model_name = "imvladikon/het5_small_summarization"
 model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 summarizer = SummarizationPipeline(model=model, tokenizer=tokenizer)

 from transformers import AutoModelForSeq2SeqLM, AutoTokenizer, SummarizationPipeline
+model_name = "imvladikon/het5_summarization"
 model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 summarizer = SummarizationPipeline(model=model, tokenizer=tokenizer)