Update README.md

README.md

@@ -32,13 +32,13 @@ def tag(text: str):
     text = text.strip().replace('\n', '')
     tokenized_text = tokenizer.encode(text, return_tensors="pt")
 
-    summary_ids = model.generate(tokenized_text,
+    tags_ids = model.generate(tokenized_text,
                                         num_beams=4,
                                         no_repeat_ngram_size=2,
                                         max_length=20,
                                         early_stopping=True)
 
-    output = tokenizer.decode(summary_ids[0], skip_special_tokens=True)
+    output = tokenizer.decode(tags_ids[0], skip_special_tokens=True)
     return output.split(',')
 
 tags = tag(article)