JustinLin610 commited on
Commit
69fedc9
1 Parent(s): 6cf5e8c
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -43,7 +43,8 @@ FourPoint = Tuple[Tuple[int, int], Tuple[int, int], Tuple[int, int], Tuple[int,
43
 
44
  reader = ReaderLite(gpu=True)
45
  overrides={"eval_cider": False, "beam": 5, "max_len_b": 64, "patch_image_size": 480,
46
- "orig_patch_image_size": 224, "no_repeat_ngram_size": 0, "seed": 42}
 
47
  models, cfg, task = checkpoint_utils.load_model_ensemble_and_task(
48
  utils.split_paths('checkpoints/ocr_general_clean.pt'),
49
  arg_overrides=overrides
 
43
 
44
  reader = ReaderLite(gpu=True)
45
  overrides={"eval_cider": False, "beam": 5, "max_len_b": 64, "patch_image_size": 480,
46
+ "orig_patch_image_size": 224, "interpolate_position": True,
47
+ "no_repeat_ngram_size": 0, "seed": 42}
48
  models, cfg, task = checkpoint_utils.load_model_ensemble_and_task(
49
  utils.split_paths('checkpoints/ocr_general_clean.pt'),
50
  arg_overrides=overrides