PARTHA181098 commited on
Commit
2284e96
·
verified ·
1 Parent(s): b7f8d18

Update agents/document_agent.py

Browse files
Files changed (1) hide show
  1. agents/document_agent.py +1 -1
agents/document_agent.py CHANGED
@@ -20,7 +20,7 @@ import io
20
  from PIL import Image
21
  # from pdf2image import convert_from_path
22
  # from transformers import DonutProcessor, VisionEncoderDecoderModel
23
- import torch
24
  from agents.base_agent import BaseAgent
25
  from state import (
26
  InvoiceProcessingState, InvoiceData, ItemDetail,
 
20
  from PIL import Image
21
  # from pdf2image import convert_from_path
22
  # from transformers import DonutProcessor, VisionEncoderDecoderModel
23
+ # import torch
24
  from agents.base_agent import BaseAgent
25
  from state import (
26
  InvoiceProcessingState, InvoiceData, ItemDetail,