Xianbao QIAN commited on
Commit
6f60561
1 Parent(s): 7c18ecd

update wording

Browse files
Files changed (1) hide show
  1. app.py +5 -5
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 the model on HF hub.
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
- # TODO: Add a dropbox for model selection
 
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()