Update README.md

README.md

@@ -13,7 +13,46 @@ datasets:
 language:
 - en
 library_name: transformers
-base_model:
-- Falconsai/text_summarization
-- meta-llama/Meta-Llama-3.1-8B-Instruct
----
+---
+import torch
+from transformers import BartForConditionalGeneration, BartTokenizer
+
+class AdvancedSummarizer:
+    def __init__(self, model_name="facebook/bart-large-cnn"):
+        self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+        self.model = BartForConditionalGeneration.from_pretrained(model_name).to(self.device)
+        self.tokenizer = BartTokenizer.from_pretrained(model_name)
+
+    def summarize(self, text, max_length=150, min_length=50, length_penalty=2.0, num_beams=4):
+        inputs = self.tokenizer([text], max_length=1024, return_tensors="pt", truncation=True)
+        inputs = inputs.to(self.device)
+
+        summary_ids = self.model.generate(
+            inputs["input_ids"],
+            num_beams=num_beams,
+            max_length=max_length,
+            min_length=min_length,
+            length_penalty=length_penalty
+        )
+
+        summary = self.tokenizer.decode(summary_ids[0], skip_special_tokens=True)
+        return summary
+
+def main():
+    # Example usage
+    summarizer = AdvancedSummarizer()
+
+    text = """
+    Artificial intelligence (AI) is intelligence demonstrated by machines, as opposed to natural intelligence displayed by animals including humans. AI research has been defined as the field of study of intelligent agents, which refers to any system that perceives its environment and takes actions that maximize its chance of achieving its goals.
+
+    The term "artificial intelligence" had previously been used to describe machines that mimic and display "human" cognitive skills that are associated with the human mind, such as "learning" and "problem-solving". This definition has since been rejected by major AI researchers who now describe AI in terms of rationality and acting rationally, which does not limit how intelligence can be articulated.
+
+    AI applications include advanced web search engines, recommendation systems, understanding human speech, self-driving cars, automated decision-making and competing at the highest level in strategic game systems. As machines become increasingly capable, tasks considered to require "intelligence" are often removed from the definition of AI, a phenomenon known as the AI effect. For instance, optical character recognition is frequently excluded from things considered to be AI, having become a routine technology.
+    """
+
+    summary = summarizer.summarize(text)
+    print("Summary:")
+    print(summary)
+
+if __name__ == "__main__":
+    main()