Update README.md

README.md

@@ -14,28 +14,36 @@ Use the code below to get started with the model.
 
 from transformers import T5ForConditionalGeneration, T5Tokenizer
 
-# Load the model and tokenizer from Hugging Face repository
+# Load the model and tokenizer from your Hugging Face repository
 
 model = T5ForConditionalGeneration.from_pretrained("Vijayendra/T5-Base-Sum")
 tokenizer = T5Tokenizer.from_pretrained("Vijayendra/T5-Base-Sum")
 
-# Example of using the model for summarization
+# Example of a random article (can replace this with any article)
 
-article = """
-The tower is 324 metres (1,063 ft) tall, about the same height as an 81-storey building, and the tallest structure in Paris. Its base is square, measuring 
-125 metres (410 ft) on each side. During its construction, the Eiffel Tower surpassed the Washington Monument to become the tallest man-made structure in the 
-world, a title it held for 41 years until the Chrysler Building in New York City was finished in 1930. It was the first structure to reach a height of 300 
-metres. Due to the addition of a broadcasting aerial at the top of the tower in 1957, it is now taller than the Chrysler Building by 5.2 metres (17 ft). 
-Excluding transmitters, it is the second tallest free-standing structure in France after the Millau Viaduct.
+random_article = """
+Videos that say approved vaccines are dangerous and cause autism, cancer or infertility are among those that will be taken down, the company said.  The policy includes the 
+termination of accounts of anti-vaccine influencers.  Tech giants have been criticised for not doing more to counter false health information on their sites.  In July, US President
+ Joe Biden said social media platforms were largely responsible for people's scepticism in getting vaccinated by spreading misinformation, and appealed for them to address the 
+ issue.  YouTube, which is owned by Google, said 130,000 videos were removed from its platform since last year, when it implemented a ban on content spreading misinformation 
+ about Covid vaccines.  In a blog post, the company said it had seen false claims about Covid jabs "spill over into misinformation about vaccines in general". The new policy 
+ covers long-approved vaccines, such as those against measles or hepatitis B.  "We're expanding our medical misinformation policies on YouTube with new guidelines on currently 
+ administered vaccines that are approved and confirmed to be safe and effective by local health authorities and the WHO," the post said, referring to the World Health Organization.
 """
-inputs = tokenizer.encode("summarize: " + article, return_tensors="pt", max_length=512, truncation=True)
-summary_ids = model.generate(inputs, max_length=150, min_length=50, length_penalty=2.0, num_beams=4, early_stopping=True)
+
+# Tokenize the input article
+
+inputs = tokenizer.encode("summarize: " + random_article, return_tensors="pt", max_length=512, truncation=True)
+
+# Generate summary
+
+summary_ids = model.generate(inputs, max_length=150, min_length=100, length_penalty=3.0, num_beams=7, early_stopping=False)
 
 # Decode and print the summary
 
 summary = tokenizer.decode(summary_ids[0], skip_special_tokens=True)
+print("Summary:")
 print(summary)
 
-
 [More Information Needed]