napatswift commited on
Commit
a73ee78
·
1 Parent(s): 1c727d5

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -5
main.py CHANGED
@@ -1,6 +1,5 @@
1
  from mmocr.apis import MMOCRInferencer
2
 
3
- from mmocr.ocr import MMOCR
4
  import gradio as gr
5
  import cv2
6
  import sys
@@ -8,11 +7,9 @@ import torch
8
 
9
  device = 'gpu' if torch.cuda.is_available() else 'cpu'
10
 
11
- ocr = MMOCR(det_config='model/det/config.py',
12
  det_ckpt='model/det/model.pth',
13
- recog_config='model/recog/config.py',
14
- recog_ckpt='model/det/model.pth',
15
- device='cpu')
16
 
17
  def get_rec(points):
18
  xs = []
 
1
  from mmocr.apis import MMOCRInferencer
2
 
 
3
  import gradio as gr
4
  import cv2
5
  import sys
 
7
 
8
  device = 'gpu' if torch.cuda.is_available() else 'cpu'
9
 
10
+ ocr = MMOCRInferencer(det_config='model/det/config.py',
11
  det_ckpt='model/det/model.pth',
12
+ device=device)
 
 
13
 
14
  def get_rec(points):
15
  xs = []