File size: 491 Bytes
d3ff6da
 
 
 
 
 
378b1f2
 
d3ff6da
6f60561
d3ff6da
 
6f60561
d3ff6da
6f60561
d3ff6da
 
6f60561
 
d3ff6da
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
import mmocr
import gradio as gr
import os
from mmocr.utils.ocr import MMOCR


config_dir = os.path.join(
    os.path.dirname(__file__), 'configs/')

# TODO: Put more models on HF hub.
ocr = MMOCR(config_dir=config_dir)


def infer(image):
  # TODO: Also display bounding boxes interactively
  return ocr.readtext(image, output='.', print_result=True, imshow=False)


# TODO: a drop down list for model selection
iface = gr.Interface(fn=infer, inputs="image", outputs="json")
iface.launch()