chandan06 commited on
Commit
cbe95d0
·
verified ·
1 Parent(s): f11bcc1

Update donut_inference.py

Browse files
Files changed (1) hide show
  1. donut_inference.py +1 -0
donut_inference.py CHANGED
@@ -10,6 +10,7 @@ load_dotenv()
10
  # imgae = image.resize((1864, 1440))
11
 
12
  device = "cuda" if torch.cuda.is_available() else "cpu"
 
13
  # Load the processor from the local directory
14
  processor = DonutProcessor.from_pretrained("Henge-navuuu/donut-base-finetuned-forms-v1")
15
 
 
10
  # imgae = image.resize((1864, 1440))
11
 
12
  device = "cuda" if torch.cuda.is_available() else "cpu"
13
+ device = "cpu"
14
  # Load the processor from the local directory
15
  processor = DonutProcessor.from_pretrained("Henge-navuuu/donut-base-finetuned-forms-v1")
16