isLinXu commited on
Commit
215f810
·
1 Parent(s): 0eb5f90
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -60,8 +60,8 @@ def ocr_inference(inputs, out_dir, det, det_weights, rec, rec_weights, kie, kie_
60
  print("call_args", call_args)
61
  ocr = MMOCRInferencer(**init_args)
62
  ocr(**call_args)
63
- save_vis_dir = '../../results/vis/'
64
- save_pred_dir = '../../results/preds/'
65
  img_out = PIL.Image.open(os.path.join(save_vis_dir, img_path))
66
  json_out = json.load(open(os.path.join(save_pred_dir, img_path.replace('.jpg', '.json'))))
67
  return img_out, json_out
 
60
  print("call_args", call_args)
61
  ocr = MMOCRInferencer(**init_args)
62
  ocr(**call_args)
63
+ save_vis_dir = './results/vis/'
64
+ save_pred_dir = './results/preds/'
65
  img_out = PIL.Image.open(os.path.join(save_vis_dir, img_path))
66
  json_out = json.load(open(os.path.join(save_pred_dir, img_path.replace('.jpg', '.json'))))
67
  return img_out, json_out