Update README.md

README.md

@@ -101,7 +101,6 @@ I've tested several other news headline generators on Hugging Face and across th
 ## How to use?
 
 ```python
-# Load model directly
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
 tokenizer = AutoTokenizer.from_pretrained("Ateeqq/news-title-generator")
@@ -110,22 +109,14 @@ model = AutoModelForSeq2SeqLM.from_pretrained("Ateeqq/news-title-generator")
 ```python
 def generate_title(input_text):
 
-  # Encode the input text
   input_ids = tokenizer.encode(input_text, return_tensors="pt")
-
-  # Generate the output
   output = model.generate(input_ids)
-
-  # Decode the generated tokens
   decoded_text = tokenizer.decode(output[0], skip_special_tokens=True)
-
   return decoded_text
 
-# Example usage
 input_text = "A group of scientists discovered a new planet."
 generated_title = generate_title(input_text)
 
-print(f"Input Text: {input_text}")
 print(f"Generated Title: {generated_title}")
 ```