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

fix: use a well trained checkpoint on sample screenshots used in playspace

Browse files
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@commit/116838e7868170d4ecc682aca4afc317b08e6009"
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"
10
 
11
  processor = DonutProcessor.from_pretrained(pretrained_repo_name)
12
  model = VisionEncoderDecoderModel.from_pretrained(pretrained_repo_name)