Update README.md

README.md

@@ -21,8 +21,39 @@ This model is based on the Facebook BART (Bidirectional and Auto-Regressive Tran
 ### Installation:
 
 You can install the necessary libraries using pip:
-  ```bash
-    pip install transformers
-    pip datasets
-    pip evaluate
-    pip rouge_score
+```bash
+pip install transformers
+```
+### Inferecnce
+```python
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+
+tokenizer = AutoTokenizer.from_pretrained("suriya7/text_summarize")
+model = AutoModelForSeq2SeqLM.from_pretrained("suriya7/text_summarize")
+
+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."
+summary = generate_summary(text_to_summarize)
+print(summary)
+```
+
+```
+Cristiano Ronaldo dos Santos Aveiro => person
+5 February 1985 => date
+Saudi Pro League => competitions
+Al Nassr => teams
+Portugal national team => teams
+Ballon d'Or => award
+UEFA Men's Player of the Year Awards => award
+European Golden Shoes => award
+UEFA Champions Leagues => competitions
+UEFA European Championship => competitions
+UEFA Nations League => competitions
+Champions League => competitions
+European Championship => competitions
+```