ivelin commited on
Commit
9cb2224
1 Parent(s): 9fbca7c

fix: switch model checkpoint

Browse files

Signed-off-by: ivelin <ivelin.eth@gmail.com>

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ import torch
6
  import html
7
  from transformers import DonutProcessor, VisionEncoderDecoderModel
8
 
9
- pretrained_repo_name = "ivelin/donut-refexp-draft"
10
 
11
  processor = DonutProcessor.from_pretrained(pretrained_repo_name)
12
  model = VisionEncoderDecoderModel.from_pretrained(pretrained_repo_name)
 
6
  import html
7
  from transformers import DonutProcessor, VisionEncoderDecoderModel
8
 
9
+ pretrained_repo_name = "ivelin/donut-refexp-draft-precision2decs"
10
 
11
  processor = DonutProcessor.from_pretrained(pretrained_repo_name)
12
  model = VisionEncoderDecoderModel.from_pretrained(pretrained_repo_name)