Spaces:
Running
Running
ivelin
commited on
Commit
•
9cb2224
1
Parent(s):
9fbca7c
fix: switch model checkpoint
Browse filesSigned-off-by: ivelin <ivelin.eth@gmail.com>
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)
|