Spaces:
Sleeping
Sleeping
Update agents/document_agent.py
Browse files- 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,
|