Update README.md
Browse files
README.md
CHANGED
@@ -57,8 +57,8 @@ import re
|
|
57 |
# Load model and processor
|
58 |
print("Loading Donut model...")
|
59 |
|
60 |
-
processor = DonutProcessor.from_pretrained("pandafm/
|
61 |
-
model = VisionEncoderDecoderModel.from_pretrained("pandafm/
|
62 |
|
63 |
if torch.cuda.is_available():
|
64 |
device = torch.device("cuda")
|
|
|
57 |
# Load model and processor
|
58 |
print("Loading Donut model...")
|
59 |
|
60 |
+
processor = DonutProcessor.from_pretrained("pandafm/donut-es")
|
61 |
+
model = VisionEncoderDecoderModel.from_pretrained("pandafm/donut-es")
|
62 |
|
63 |
if torch.cuda.is_available():
|
64 |
device = torch.device("cuda")
|