Hugging Face's logo

Spaces:
ivelin
/
ui-refexp
Running

App Files Community
2
ivelin commited on
Commit
0bf3b92
•
1 Parent(s): 8c75a6c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ pretrained_repo_name = "ivelin/donut-refexp-combined-v1"
10
  print(f"Loading model checkpoint: {pretrained_repo_name}")
11
 
12
  processor = DonutProcessor.from_pretrained(pretrained_repo_name)
13
- model = VisionEncoderDecoderModel.from_pretrained(pretrained_repo_name, revision="9c24fb7cb285de7f1f299d4fbfee2bacb46c07f6")
14
 
15
  device = "cuda" if torch.cuda.is_available() else "cpu"
16
  model.to(device)
 
10
  print(f"Loading model checkpoint: {pretrained_repo_name}")
11
 
12
  processor = DonutProcessor.from_pretrained(pretrained_repo_name)
13
+ model = VisionEncoderDecoderModel.from_pretrained(pretrained_repo_name)
14
 
15
  device = "cuda" if torch.cuda.is_available() else "cpu"
16
  model.to(device)