Update main.py

main.py

@@ -1,6 +1,5 @@
 from mmocr.apis import MMOCRInferencer
 
-from mmocr.ocr import MMOCR
 import gradio as gr
 import cv2
 import sys
@@ -8,11 +7,9 @@ import torch
 
 device = 'gpu' if torch.cuda.is_available() else 'cpu'
 
-ocr = MMOCR(det_config='model/det/config.py',
+ocr = MMOCRInferencer(det_config='model/det/config.py',
             det_ckpt='model/det/model.pth',
-            recog_config='model/recog/config.py',
-            recog_ckpt='model/det/model.pth',
-            device='cpu')
+            device=device)
 
 def get_rec(points):
     xs = []