Spaces:

OFA-Sys
/

OFA-OCR

Runtime error

JustinLin610 commited on Nov 16, 2022

Commit

aa287a2

•

1 Parent(s): fb33609

add ckpt

Files changed (1) hide show

app.py CHANGED Viewed

@@ -116,7 +116,7 @@ reader = ReaderLite()
 overrides={"eval_cider": False, "beam": 8, "max_len_b": 128, "patch_image_size": 480,
            "orig_patch_image_size": 224, "no_repeat_ngram_size": 0, "seed": 7}
 models, cfg, task = checkpoint_utils.load_model_ensemble_and_task(
-    utils.split_paths('checkpoints/ocr.pt'),
     arg_overrides=overrides
 )

 overrides={"eval_cider": False, "beam": 8, "max_len_b": 128, "patch_image_size": 480,
            "orig_patch_image_size": 224, "no_repeat_ngram_size": 0, "seed": 7}
 models, cfg, task = checkpoint_utils.load_model_ensemble_and_task(
+    utils.split_paths('checkpoints/ocr_general_clean.pt'),
     arg_overrides=overrides
 )