Spaces:
Running
Running
Update pdf_processor.py
Browse files- pdf_processor.py +2 -1
pdf_processor.py
CHANGED
@@ -26,7 +26,8 @@ class PDFProcessor:
|
|
26 |
model="facebook/bart-large-cnn",
|
27 |
device_map="auto", # Automatically choose best device
|
28 |
torch_dtype=torch.float32, # Use float32 for better memory efficiency
|
29 |
-
batch_size=1
|
|
|
30 |
)
|
31 |
self.progress_callback = None
|
32 |
|
|
|
26 |
model="facebook/bart-large-cnn",
|
27 |
device_map="auto", # Automatically choose best device
|
28 |
torch_dtype=torch.float32, # Use float32 for better memory efficiency
|
29 |
+
batch_size=1 , # Process one chunk at a time
|
30 |
+
trust_remote_code=True
|
31 |
)
|
32 |
self.progress_callback = None
|
33 |
|