Update README.md

README.md

@@ -8,15 +8,10 @@ pipeline_tag: summarization
 from transformers import AutoModelForSeq2SeqLM, AutoTokenizer, SummarizationPipeline
 
 
-class HebrewSummarizationPipeline(SummarizationPipeline):
-    
-    @classmethod
-    def from_pretrained(self, *args, **kwargs):
-        model = AutoModelForSeq2SeqLM.from_pretrained(*args, **kwargs)
-        tokenizer = AutoTokenizer.from_pretrained(*args, **kwargs)
-        return self(model=model, tokenizer=tokenizer)
-
-summarizer = HebrewSummarizationPipeline.from_pretrained("imvladikon/het5_small_summarization")
+model_name = "imvladikon/het5_small_summarization"
+model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+summarizer = SummarizationPipeline(model=model, tokenizer=tokenizer)
 ```      
      
 example