Hugging Face's logo

Spaces:
ivelin
/
ui-refexp
Runtime error

App Files Community
2
ivelin commited on
Commit
6b8a041
·
1 Parent(s): 81c6fa4

update with improved model checkpoint

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -7,7 +7,8 @@ import html
7
  from transformers import DonutProcessor, VisionEncoderDecoderModel
8
 
9
  pretrained_repo_name = 'ivelin/donut-refexp-combined-v1'
10
- pretrained_revision = '41210d7c42a22e77711711ec45508a6b63ec380f' # : IoU=0.42 # 'main' for latest revision
 
11
  print(f"Loading model checkpoint: {pretrained_repo_name}")
12
 
13
  processor = DonutProcessor.from_pretrained(pretrained_repo_name, revision=pretrained_revision)
 
7
  from transformers import DonutProcessor, VisionEncoderDecoderModel
8
 
9
  pretrained_repo_name = 'ivelin/donut-refexp-combined-v1'
10
+ pretrained_revision = '348ddad8e958d370b7e341acd6050330faa0500f' # Iou = 0.47
11
+ # '41210d7c42a22e77711711ec45508a6b63ec380f' # : IoU=0.42 # 'main' for latest revision
12
  print(f"Loading model checkpoint: {pretrained_repo_name}")
13
 
14
  processor = DonutProcessor.from_pretrained(pretrained_repo_name, revision=pretrained_revision)