asalhi85 commited on
Commit
3be2994
1 Parent(s): 3b1e679

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +132 -0
app.py ADDED
@@ -0,0 +1,132 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import pandas as pd
2
+ import PIL
3
+ from PIL import Image
4
+ from PIL import ImageDraw
5
+ import gradio as gr
6
+ import torch
7
+ import easyocr
8
+
9
+ torch.hub.download_url_to_file('https://github.com/JaidedAI/EasyOCR/raw/master/examples/english.png', 'english.png')
10
+ torch.hub.download_url_to_file('https://github.com/JaidedAI/EasyOCR/raw/master/examples/thai.jpg', 'thai.jpg')
11
+ torch.hub.download_url_to_file('https://github.com/JaidedAI/EasyOCR/raw/master/examples/french.jpg', 'french.jpg')
12
+ torch.hub.download_url_to_file('https://github.com/JaidedAI/EasyOCR/raw/master/examples/chinese.jpg', 'chinese.jpg')
13
+ torch.hub.download_url_to_file('https://github.com/JaidedAI/EasyOCR/raw/master/examples/japanese.jpg', 'japanese.jpg')
14
+ torch.hub.download_url_to_file('https://github.com/JaidedAI/EasyOCR/raw/master/examples/korean.png', 'korean.png')
15
+ torch.hub.download_url_to_file('https://i.imgur.com/mwQFd7G.jpeg', 'Hindi.jpeg')
16
+
17
+ def draw_boxes(image, bounds, color='yellow', width=2):
18
+ draw = ImageDraw.Draw(image)
19
+ for bound in bounds:
20
+ p0, p1, p2, p3 = bound[0]
21
+ draw.line([*p0, *p1, *p2, *p3, *p0], fill=color, width=width)
22
+ return image
23
+
24
+ def inference(img, lang):
25
+ reader = easyocr.Reader(lang)
26
+ bounds = reader.readtext(img.name)
27
+ im = PIL.Image.open(img.name)
28
+ draw_boxes(im, bounds)
29
+ im.save('result.jpg')
30
+ return ['result.jpg', pd.DataFrame(bounds).iloc[: , 1:]]
31
+
32
+ title = 'EasyOCR'
33
+ description = 'Gradio demo for EasyOCR. EasyOCR demo supports 80+ languages.To use it, simply upload your image and choose a language from the dropdown menu, or click one of the examples to load them. Read more at the links below.'
34
+ article = "<p style='text-align: center'><a href='https://www.jaided.ai/easyocr/'>Ready-to-use OCR with 80+ supported languages and all popular writing scripts including Latin, Chinese, Arabic, Devanagari, Cyrillic and etc.</a> | <a href='https://github.com/JaidedAI/EasyOCR'>Github Repo</a></p>"
35
+ examples = [['english.png',['en']],['thai.jpg',['th']],['french.jpg',['fr', 'en']],['chinese.jpg',['ch_sim', 'en']],['japanese.jpg',['ja', 'en']],['korean.png',['ko', 'en']],['Hindi.jpeg',['hi', 'en']]]
36
+ css = ".output_image, .input_image {height: 40rem !important; width: 100% !important;}"
37
+ choices = [
38
+ "abq",
39
+ "ady",
40
+ "af",
41
+ "ang",
42
+ "ar",
43
+ "as",
44
+ "ava",
45
+ "az",
46
+ "be",
47
+ "bg",
48
+ "bh",
49
+ "bho",
50
+ "bn",
51
+ "bs",
52
+ "ch_sim",
53
+ "ch_tra",
54
+ "che",
55
+ "cs",
56
+ "cy",
57
+ "da",
58
+ "dar",
59
+ "de",
60
+ "en",
61
+ "es",
62
+ "et",
63
+ "fa",
64
+ "fr",
65
+ "ga",
66
+ "gom",
67
+ "hi",
68
+ "hr",
69
+ "hu",
70
+ "id",
71
+ "inh",
72
+ "is",
73
+ "it",
74
+ "ja",
75
+ "kbd",
76
+ "kn",
77
+ "ko",
78
+ "ku",
79
+ "la",
80
+ "lbe",
81
+ "lez",
82
+ "lt",
83
+ "lv",
84
+ "mah",
85
+ "mai",
86
+ "mi",
87
+ "mn",
88
+ "mr",
89
+ "ms",
90
+ "mt",
91
+ "ne",
92
+ "new",
93
+ "nl",
94
+ "no",
95
+ "oc",
96
+ "pi",
97
+ "pl",
98
+ "pt",
99
+ "ro",
100
+ "ru",
101
+ "rs_cyrillic",
102
+ "rs_latin",
103
+ "sck",
104
+ "sk",
105
+ "sl",
106
+ "sq",
107
+ "sv",
108
+ "sw",
109
+ "ta",
110
+ "tab",
111
+ "te",
112
+ "th",
113
+ "tjk",
114
+ "tl",
115
+ "tr",
116
+ "ug",
117
+ "uk",
118
+ "ur",
119
+ "uz",
120
+ "vi"
121
+ ]
122
+ gr.Interface(
123
+ inference,
124
+ [gr.inputs.Image(type='file', label='Input'),gr.inputs.CheckboxGroup(choices, type="value", default=['en'], label='language')],
125
+ [gr.outputs.Image(type='file', label='Output'), gr.outputs.Dataframe(headers=['text', 'confidence'])],
126
+ title=title,
127
+ description=description,
128
+ article=article,
129
+ examples=examples,
130
+ css=css,
131
+ enable_queue=True
132
+ ).launch(debug=True)