Update README.md

README.md

@@ -54,7 +54,7 @@ tokenizer = AutoTokenizer.from_pretrained(model_name)
 input_text = "Your paragraph here..."
 inputs = tokenizer(input_text, return_tensors="pt")
 outputs = model.generate(**inputs, max_length=512)
-decoded_output = tokenizer.decode(outputs, skip_special_tokens=True)
+decoded_output = tokenizer.decode(outputs[0], skip_special_tokens=True)
 
 # Process the output to get a list of keywords (split and remove duplicates)
 keywords = list(set(decoded_output.split('||')))