DerrylNessie commited on
Commit
334adc2
1 Parent(s): 1877ef8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +115 -23
app.py CHANGED
@@ -1,36 +1,128 @@
1
- import os
2
- os.system('pip install paddlepaddle')
3
- os.system('pip install paddleocr')
4
- from paddleocr import PaddleOCR, draw_ocr
5
  from PIL import Image
 
6
  import gradio as gr
7
  import torch
 
8
 
9
- torch.hub.download_url_to_file('https://i.imgur.com/aqMBT0i.jpg', 'example.jpg')
 
 
 
 
 
 
 
 
 
 
 
 
 
10
 
11
  def inference(img, lang):
12
- ocr = PaddleOCR(use_angle_cls=True, lang=lang,use_gpu=False)
13
- img_path = img.name
14
- result = ocr.ocr(img_path, cls=True)
15
- image = Image.open(img_path).convert('RGB')
16
- boxes = [line[0] for line in result]
17
- txts = [line[1][0] for line in result]
18
- scores = [line[1][1] for line in result]
19
- im_show = draw_ocr(image, boxes, txts, scores,
20
- font_path='simfang.ttf')
21
- im_show = Image.fromarray(im_show)
22
- im_show.save('result.jpg')
23
- return 'result.jpg'
24
 
25
- title = 'PaddleOCR'
26
- description = 'Gradio demo for PaddleOCR. PaddleOCR demo supports Chinese, English, French, German, Korean and Japanese.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.'
27
- article = "<p style='text-align: center'><a href='https://www.paddlepaddle.org.cn/hub/scene/ocr'>Awesome multilingual OCR toolkits based on PaddlePaddle (practical ultra lightweight OCR system, support 80+ languages recognition, provide data annotation and synthesis tools, support training and deployment among server, mobile, embedded and IoT devices)</a> | <a href='https://github.com/PaddlePaddle/PaddleOCR'>Github Repo</a></p>"
28
- examples = [['example.jpg','en']]
29
  css = ".output_image, .input_image {height: 40rem !important; width: 100% !important;}"
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
30
  gr.Interface(
31
  inference,
32
- [gr.inputs.Image(type='file', label='Input'),gr.inputs.Dropdown(choices=['ch', 'en', 'fr', 'german', 'korean', 'japan'], type="value", default='en', label='language')],
33
- gr.outputs.Image(type='file', label='Output'),
34
  title=title,
35
  description=description,
36
  article=article,
 
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,