Geewook Kim
commited on
Commit
•
8a12930
1
Parent(s):
a43f1aa
Update app.py
Browse files
app.py
CHANGED
@@ -47,8 +47,8 @@ if __name__ == "__main__":
|
|
47 |
example_sample = []
|
48 |
if args.sample_img_path:
|
49 |
image = Image.open(args.sample_img_path)
|
50 |
-
image.save(
|
51 |
-
example_sample.append(
|
52 |
|
53 |
pretrained_model = DonutModel.from_pretrained(args.pretrained_path)
|
54 |
pretrained_model.encoder.to(torch.bfloat16)
|
|
|
47 |
example_sample = []
|
48 |
if args.sample_img_path:
|
49 |
image = Image.open(args.sample_img_path)
|
50 |
+
image.save("cord_sample_receipt.png")
|
51 |
+
example_sample.append("cord_sample_receipt.png")
|
52 |
|
53 |
pretrained_model = DonutModel.from_pretrained(args.pretrained_path)
|
54 |
pretrained_model.encoder.to(torch.bfloat16)
|