Update app.py

app.py

@@ -196,11 +196,11 @@ try:
         spacy.cli.download("en_core_web_sm")
         nlp = spacy.load("en_core_web_sm")
     print("✅ Loading NLP models...")
-    from transformers import PegasusTokenizer
+    # Use Facebook's bart-large-cnn for summarization
     summarizer = pipeline(
         "summarization",
-        model="nsi319/legal-pegasus",
-        tokenizer=PegasusTokenizer.from_pretrained("nsi319/legal-pegasus", use_fast=False),
+        model="facebook/bart-large-cnn",
+        tokenizer="facebook/bart-large-cnn",
         device=0 if torch.cuda.is_available() else -1
     )
     # Optionally convert summarizer model to FP16 for faster inference on GPU