Spaces:
Running
Running
Create app.py
Browse files
app.py
ADDED
@@ -0,0 +1,36 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# -*- coding: utf-8 -*-
|
2 |
+
"""
|
3 |
+
@author:XuMing([email protected])
|
4 |
+
@description: pip install gradio
|
5 |
+
"""
|
6 |
+
|
7 |
+
import gradio as gr
|
8 |
+
from imgocr import ImgOcr
|
9 |
+
model = ImgOcr()
|
10 |
+
|
11 |
+
|
12 |
+
def get_text(img_path):
|
13 |
+
ocr_result = model.ocr(img_path)[0]
|
14 |
+
print("{} \t\t {}".format(img_path, ocr_result))
|
15 |
+
ocr_text = [i[-1][0] for i in ocr_result]
|
16 |
+
r = '\n'.join(ocr_text)
|
17 |
+
print(r)
|
18 |
+
return r
|
19 |
+
|
20 |
+
|
21 |
+
if __name__ == '__main__':
|
22 |
+
examples = [
|
23 |
+
['data/1.jpg'],
|
24 |
+
['data/11.jpg'],
|
25 |
+
['data/12.jpg'],
|
26 |
+
['data/00111002.jpg'],
|
27 |
+
]
|
28 |
+
gr.Interface(
|
29 |
+
get_text,
|
30 |
+
inputs=['image'],
|
31 |
+
outputs='text',
|
32 |
+
title="imgocr: Image OCR",
|
33 |
+
description="Input an image, and the machine will output the OCR result.",
|
34 |
+
article="Link to <a href='https://github.com/shibing624/imgocr' style='color:blue;' target='_blank\'>Github REPO</a>",
|
35 |
+
examples=examples
|
36 |
+
).launch()
|