Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -20,7 +20,7 @@ image.save("cord_sample_receipt1.png")
|
|
20 |
image = Image.open("./sample_image_cord_test_receipt_00012.png")
|
21 |
image.save("cord_sample_receipt2.png")
|
22 |
|
23 |
-
pretrained_model = DonutModel.from_pretrained("Raj-Master/donut-demo-123")
|
24 |
pretrained_model.encoder.to(torch.bfloat16)
|
25 |
pretrained_model.eval()
|
26 |
|
|
|
20 |
image = Image.open("./sample_image_cord_test_receipt_00012.png")
|
21 |
image.save("cord_sample_receipt2.png")
|
22 |
|
23 |
+
pretrained_model = DonutModel.from_pretrained("Raj-Master/donut-demo-123/tree/main")
|
24 |
pretrained_model.encoder.to(torch.bfloat16)
|
25 |
pretrained_model.eval()
|
26 |
|