Update README.md

README.md

@@ -19,22 +19,22 @@ This model is based on the Facebook BART (Bidirectional and Auto-Regressive Tran
 ### Inference
 
 ```python
- from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
- tokenizer = AutoTokenizer.from_pretrained("suriya7/text_summarize")
- model = AutoModelForSeq2SeqLM.from_pretrained("suriya7/text_summarize")
+tokenizer = AutoTokenizer.from_pretrained("suriya7/text_summarize")
+model = AutoModelForSeq2SeqLM.from_pretrained("suriya7/text_summarize")
 
- def generate_summary(text):
+def generate_summary(text):
 
    inputs = tokenizer([text], max_length=1024, return_tensors='pt', truncation=True)
-
+  
    summary_ids = model.generate(inputs['input_ids'],max_new_tokens=100, do_sample=False)
-
+  
    summary = tokenizer.decode(summary_ids[0], skip_special_tokens=True)
    return summary
 
 text_to_summarize = "Now, there is no doubt that one of the most important aspects of any Pixel phone is its camera.
-                     And there might be good news for all camera lovers. Rumours have suggested  that the Pixel 9 could come with a telephoto lens,
-                     improving its photography capabilities even further. Google will likely continue to focus on using AI to
-                     enhance its camera performance, in order to make sure that Pixel phones remain top contenders in the world of mobile photography"
+                   And there might be good news for all camera lovers. Rumours have suggested  that the Pixel 9 could come with a telephoto lens,
+                   improving its photography capabilities even further. Google will likely continue to focus on using AI to
+                   enhance its camera performance, in order to make sure that Pixel phones remain top contenders in the world of mobile photography"
 summary = generate_summary(text_to_summarize)