Spaces:
Runtime error
Runtime error
Xianbao QIAN
commited on
Commit
•
6f60561
1
Parent(s):
7c18ecd
update wording
Browse files
app.py
CHANGED
@@ -1,21 +1,21 @@
|
|
1 |
import mmocr
|
2 |
import gradio as gr
|
3 |
import os
|
4 |
-
from huggingface_hub import hf_hub_url
|
5 |
from mmocr.utils.ocr import MMOCR
|
6 |
|
7 |
|
8 |
-
# The default config path is based the current woking directory.
|
9 |
config_dir = os.path.join(
|
10 |
os.path.dirname(__file__), 'configs/')
|
11 |
|
12 |
-
# TODO: Put
|
13 |
ocr = MMOCR(config_dir=config_dir)
|
14 |
|
|
|
15 |
def infer(image):
|
16 |
-
# TODO: Also display bounding boxes
|
17 |
return ocr.readtext(image, output='.', print_result=True, imshow=False)
|
18 |
|
19 |
-
|
|
|
20 |
iface = gr.Interface(fn=infer, inputs="image", outputs="json")
|
21 |
iface.launch()
|
|
|
1 |
import mmocr
|
2 |
import gradio as gr
|
3 |
import os
|
|
|
4 |
from mmocr.utils.ocr import MMOCR
|
5 |
|
6 |
|
|
|
7 |
config_dir = os.path.join(
|
8 |
os.path.dirname(__file__), 'configs/')
|
9 |
|
10 |
+
# TODO: Put more models on HF hub.
|
11 |
ocr = MMOCR(config_dir=config_dir)
|
12 |
|
13 |
+
|
14 |
def infer(image):
|
15 |
+
# TODO: Also display bounding boxes interactively
|
16 |
return ocr.readtext(image, output='.', print_result=True, imshow=False)
|
17 |
|
18 |
+
|
19 |
+
# TODO: a drop down list for model selection
|
20 |
iface = gr.Interface(fn=infer, inputs="image", outputs="json")
|
21 |
iface.launch()
|