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/blob/main")
|
| 24 |
pretrained_model.encoder.to(torch.bfloat16)
|
| 25 |
pretrained_model.eval()
|
| 26 |
|