Update app.py

app.py

@@ -193,11 +193,12 @@ try:
         nlp = spacy.load("en_core_web_sm")
     print("✅ Loading NLP models...")
 
-    # Initialize summarizer with a slow tokenizer
+    # Use the slow PegasusTokenizer explicitly
+    from transformers import PegasusTokenizer
     summarizer = pipeline(
         "summarization",
         model="nsi319/legal-pegasus",
-        tokenizer=AutoTokenizer.from_pretrained("nsi319/legal-pegasus", use_fast=False),
+        tokenizer=PegasusTokenizer.from_pretrained("nsi319/legal-pegasus", use_fast=False),
         device=0 if torch.cuda.is_available() else -1
     )
     
@@ -206,7 +207,6 @@ try:
     speech_to_text = pipeline("automatic-speech-recognition", model="openai/whisper-medium", chunk_length_s=30,
                               device_map="auto" if torch.cuda.is_available() else "cpu")
 
-    # Load or fine tune CUAD QA model
     if os.path.exists("fine_tuned_legal_qa"):
         print("✅ Loading fine-tuned CUAD QA model from fine_tuned_legal_qa...")
         cuad_tokenizer = AutoTokenizer.from_pretrained("fine_tuned_legal_qa")