EuroPython2022/mmocr-demo

1import mmocr2import gradio as gr3import os4from mmocr.utils.ocr import MMOCR5 6 7config_dir = os.path.join(8    os.path.dirname(__file__), 'configs/')9 10# TODO: Put more models on HF hub.11ocr = MMOCR(config_dir=config_dir)12 13 14def infer(image):15  # TODO: Also display bounding boxes interactively16  return ocr.readtext(image, output='.', print_result=True, imshow=False)17 18 19# TODO: a drop down list for model selection20iface = gr.Interface(fn=infer, inputs="image", outputs="json")21iface.launch()22