Update app.py

app.py

@@ -185,6 +185,9 @@ def fine_tune_cuad_model():
 #    Load NLP Models       #
 #############################
 
+# ... (other imports and code remain unchanged)
+
+# Load NLP models
 try:
     try:
         nlp = spacy.load("en_core_web_sm")
@@ -193,8 +196,13 @@ try:
         nlp = spacy.load("en_core_web_sm")
     print("✅ Loading NLP models...")
 
-    summarizer = pipeline("summarization", model="nsi319/legal-pegasus",
-                            device=0 if torch.cuda.is_available() else -1)
+    # Updated summarizer: add trust_remote_code=True to load custom model files if needed.
+    summarizer = pipeline(
+        "summarization",
+        model="nsi319/legal-pegasus",
+        trust_remote_code=True,
+        device=0 if torch.cuda.is_available() else -1
+    )
     embedding_model = SentenceTransformer("all-mpnet-base-v2", device=device)
     ner_model = pipeline("ner", model="dslim/bert-base-NER",
                          device=0 if torch.cuda.is_available() else -1)
@@ -203,6 +211,9 @@ try:
                               chunk_length_s=30,
                               device_map="auto" if torch.cuda.is_available() else "cpu")
 
+    # ... (rest of your model loading code remains unchanged)
+
+
     # 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...")