Hassankhwileh commited on
Commit
638e211
·
verified ·
1 Parent(s): 371e1b5

Update pdf_processor.py

Browse files
Files changed (1) hide show
  1. 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 # Process one chunk at a time
 
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