Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -6,15 +6,8 @@ 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='
|
18 |
draw = ImageDraw.Draw(image)
|
19 |
for bound in bounds:
|
20 |
p0, p1, p2, p3 = bound[0]
|
@@ -23,7 +16,7 @@ def draw_boxes(image, bounds, color='yellow', width=2):
|
|
23 |
|
24 |
def inference(img, lang):
|
25 |
reader = easyocr.Reader(lang)
|
26 |
-
bounds = reader.readtext(img)
|
27 |
im = PIL.Image.open(img)
|
28 |
draw_boxes(im, bounds)
|
29 |
im.save('result.jpg')
|
|
|
6 |
import torch
|
7 |
import easyocr
|
8 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
9 |
|
10 |
+
def draw_boxes(image, bounds, color='green', width=2):
|
11 |
draw = ImageDraw.Draw(image)
|
12 |
for bound in bounds:
|
13 |
p0, p1, p2, p3 = bound[0]
|
|
|
16 |
|
17 |
def inference(img, lang):
|
18 |
reader = easyocr.Reader(lang)
|
19 |
+
bounds = reader.readtext(img, paragraph = True)
|
20 |
im = PIL.Image.open(img)
|
21 |
draw_boxes(im, bounds)
|
22 |
im.save('result.jpg')
|