Spaces:

napatswift
/

votecount-ml-be

Build error

napatswift commited on Feb 24, 2023

Commit

ceecba1

1 Parent(s): 0fc4b93

Update paths

Files changed (1) hide show

main.py CHANGED Viewed

@@ -2,9 +2,9 @@ from mmocr.ocr import MMOCR
 import gradio as gr
 import cv2
-model_dir = 'work_dirs/dbnet_resnet18_fpnc_1200e_icdar2015'
 ocr = MMOCR(det='DB_r18',
-            det_config=f'{model_dir}/dbnet_resnet18_fpnc_1200e_icdar2015.py',
             det_ckpt=f'{model_dir}/epoch_40.pth'
             )

 import gradio as gr
 import cv2
+model_dir = 'model'
 ocr = MMOCR(det='DB_r18',
+            det_config=f'{model_dir}/config.py',
             det_ckpt=f'{model_dir}/epoch_40.pth'
             )