Update extract_text_from_pdf.py

extract_text_from_pdf.py

@@ -38,7 +38,7 @@ class PDFTextExtractor:
         
         # Initialize model and tokenizer
         self.accelerator = Accelerator()
-        self.model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.bfloat16,use_safetensors=True,device_map=device)
+        self.model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.bfloat16,use_safetensors=True,device_map=self.device)
         self.tokenizer = AutoTokenizer.from_pretrained(model_name, use_safetensors=True)
         self.model, self.tokenizer = self.accelerator.prepare(self.model, self.tokenizer)