ysharma HF staff commited on
Commit
0e0d5e8
1 Parent(s): 39aa976
Files changed (1) hide show
  1. app.py +6 -4
app.py CHANGED
@@ -2,15 +2,17 @@ import gradio as gr
2
 
3
  def wordsrch(img):
4
  lang = "en"
5
- ocrimg = gr.Interface.load("spaces/PaddlePaddle/PaddleOCR")(img, lang)
6
- return ocrimg
 
7
 
8
  #examples = [["wordsearch2.jpg","en"]]
9
  examples = ["wordsearch2.jpg"]
10
 
11
  iface = gr.Interface(fn=wordsrch,
12
  # Please choose from: 'numpy', 'pil', 'filepath'.
13
- inputs=gr.inputs.Image(type='numpy', label='Input'), #"en"],
14
- outputs=gr.outputs.Image(type='file', label='Output'),
 
15
  examples = examples)
16
  iface.launch(debug=True, enable_queue=True)
 
2
 
3
  def wordsrch(img):
4
  lang = "en"
5
+ #ocrimg = gr.Interface.load("spaces/PaddlePaddle/PaddleOCR")(img, lang)
6
+ ocrimg, dframe = gr.Interface.load("spaces/tomofi/EasyOCR")(img, lang)
7
+ return ocrimg, dframe
8
 
9
  #examples = [["wordsearch2.jpg","en"]]
10
  examples = ["wordsearch2.jpg"]
11
 
12
  iface = gr.Interface(fn=wordsrch,
13
  # Please choose from: 'numpy', 'pil', 'filepath'.
14
+ inputs=gr.inputs.Image(type='file', label='Input'), #"en"],
15
+ #outputs=gr.outputs.Image(type='file', label='Output'),
16
+ outputs=[gr.outputs.Image(type='file', label='Output'), gr.outputs.Dataframe(headers=['text', 'confidence'])],
17
  examples = examples)
18
  iface.launch(debug=True, enable_queue=True)