Update agents/document_agent.py

agents/document_agent.py

@@ -128,10 +128,10 @@ class DocumentAgent(BaseAgent):
         self.device = "cuda" if torch.cuda.is_available() else "cpu"
 
         self.processor = DonutProcessor.from_pretrained(
-            "naver-clova-ix/donut-base-finetuned-invoice"
+            "naver-clova-ix/donut-base-finetuned-cord-v2"
         )
         self.model = VisionEncoderDecoderModel.from_pretrained(
-            "naver-clova-ix/donut-base-finetuned-invoice"
+            "naver-clova-ix/donut-base-finetuned-cord-v2"
         ).to(self.device)