update ckpt

app.py

@@ -24,8 +24,8 @@ import gradio as gr
 # Register refcoco task
 tasks.register_task('ocr', OcrTask)
 
-os.system('wget http://xc-models.oss-cn-zhangjiakou.aliyuncs.com/ofa/chinese/ocr/general/checkpoint_last.pt; '
-          'mkdir -p checkpoints; mv checkpoint_last.pt checkpoints/ocr.pt')
+os.system('wget https://shuangqing-multimodal.oss-cn-zhangjiakou.aliyuncs.com/ocr_general.pt; '
+          'mkdir -p checkpoints; mv ocr_general.pt checkpoints/ocr.pt')
 
 # turn on cuda if GPU is available
 use_cuda = torch.cuda.is_available()
@@ -112,8 +112,8 @@ def patch_resize_transform(patch_image_size=480, is_document=False):
 
 
 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}
+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