Spaces:

jinhybr
/

OCR-Receipt-Donut-Demo

Runtime error

jinhybr commited on Nov 3, 2022

Commit

9b87379

•

1 Parent(s): d946f0f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,8 +4,8 @@ import torch
 from transformers import DonutProcessor, VisionEncoderDecoderModel
-processor = DonutProcessor.from_pretrained("jinhybr/donut-base-sroie")
-model = VisionEncoderDecoderModel.from_pretrained("jinhybr/donut-base-sroie")
 device = "cuda" if torch.cuda.is_available() else "cpu"
 model.to(device)

 from transformers import DonutProcessor, VisionEncoderDecoderModel
+processor = DonutProcessor.from_pretrained("naver-clova-ix/donut-base-finetuned-cord-v2")
+model = VisionEncoderDecoderModel.from_pretrained("naver-clova-ix/donut-base-finetuned-cord-v2")
 device = "cuda" if torch.cuda.is_available() else "cpu"
 model.to(device)