vishnun commited on
Commit
82871cf
β€’
1 Parent(s): d4e28b3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -1,11 +1,11 @@
1
  import gradio as gr
2
  from craft_hw_ocr import OCR
3
 
4
- trocr = OCR.load_TrOCRmodel()
5
 
6
  def do_ocr(inp):
7
- img, results = OCR.craft_detection(inp)
8
- bboxes, text = OCR.text_recoginition(img, results, trocr[0], trocr[1])
9
  return OCR.visualize(img, results), text
10
 
11
  inputs = gr.inputs.Image()
1
  import gradio as gr
2
  from craft_hw_ocr import OCR
3
 
4
+ ocr = OCR.load_models()
5
 
6
  def do_ocr(inp):
7
+ img, results = OCR.detection(inp, ocr[2])
8
+ bboxes, text = OCR.recoginition(img, results, ocr[0], ocr[1])
9
  return OCR.visualize(img, results), text
10
 
11
  inputs = gr.inputs.Image()