Hugging Face's logo

Spaces:
ivelin
/
ui-refexp
Runtime error

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

Update app.py

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