Spaces:

Sai16216
/

PDF_Assist

Sleeping

Sai16216 commited on Nov 15, 2025

Commit

81f2565

verified ·

1 Parent(s): c5b1b11

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -193,7 +193,7 @@ def bart_summarize(text):
         truncation=True,
         max_length=4096,
     ).to(device)
-    model.config.max_length = 4096
     with torch.no_grad():
         summary_ids = bart_model.generate(
             inputs["input_ids"],

         truncation=True,
         max_length=4096,
     ).to(device)
+    bart_model.config.max_length = 4096
     with torch.no_grad():
         summary_ids = bart_model.generate(
             inputs["input_ids"],