Spaces:
Runtime error
Runtime error
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()
|