Update agents/document_agent.py

agents/document_agent.py

@@ -346,6 +346,7 @@ class DocumentAgent(BaseAgent):
 
     async def _extract_invoice_with_donut(self, file_name: str):
         images = self._pdf_to_images(file_name)
+        print('images from donut', images)
     
         # Assuming first page invoice
         image = images[0]
@@ -373,7 +374,7 @@ class DocumentAgent(BaseAgent):
         sequence = self.processor.batch_decode(outputs.sequences)[0]
         sequence = sequence.replace(self.processor.tokenizer.eos_token, "").strip()
         sequence = sequence.replace(task_prompt, "").strip()
-    
+        print('seq from donut:', sequence)
         return safe_json_parse(sequence)
 
     async def _enhance_invoice_data(self, invoice_data: InvoiceData, raw_text: str) -> InvoiceData: