Geewook Kim commited on
Commit
8a12930
1 Parent(s): a43f1aa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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(args.sample_img_path)
51
- example_sample.append(args.sample_img_path)
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)