Spaces:
Runtime error
Runtime error
isLinXu
commited on
Commit
·
215f810
1
Parent(s):
0eb5f90
update
Browse files
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 = '
|
64 |
-
save_pred_dir = '
|
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
|