Spaces:

napatswift
/

votecount-ml-be

Build error

napatswift commited on Feb 24, 2023

Commit

0ad5782

•

1 Parent(s): edc1665

Change to cpu

Files changed (1) hide show

main.py CHANGED Viewed

@@ -3,10 +3,8 @@ 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'
-            )
 def predict(inp):
     return str(ocr.readtext(inp))

 import cv2
 model_dir = 'model'
+ocr = MMOCR(det_config=f'{model_dir}/config.py',
+            det_ckpt=f'{model_dir}/epoch_40.pth', device='cpu')
 def predict(inp):
     return str(ocr.readtext(inp))