Spaces:
Runtime error
Runtime error
Ahsen Khaliq
commited on
Commit
•
6f6e791
1
Parent(s):
e67f683
Update app.py
Browse files
app.py
CHANGED
@@ -5,11 +5,11 @@ from paddleocr import PaddleOCR, draw_ocr
|
|
5 |
from PIL import Image
|
6 |
import gradio as gr
|
7 |
import torch
|
8 |
-
ocr = PaddleOCR(use_angle_cls=True, lang='en',use_gpu=False)
|
9 |
|
10 |
torch.hub.download_url_to_file('https://i.imgur.com/aqMBT0i.jpg', 'example.jpg')
|
11 |
|
12 |
-
def inference(img):
|
|
|
13 |
img_path = img.name
|
14 |
result = ocr.ocr(img_path, cls=True)
|
15 |
image = Image.open(img_path).convert('RGB')
|
@@ -23,17 +23,18 @@ def inference(img):
|
|
23 |
return 'result.jpg'
|
24 |
|
25 |
title = 'PaddleOCR'
|
26 |
-
description = 'Gradio demo for PaddleOCR. To use it, simply upload your image, 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']]
|
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'),
|
33 |
gr.outputs.Image(type='file', label='Output'),
|
34 |
title=title,
|
35 |
description=description,
|
36 |
article=article,
|
37 |
examples=examples,
|
38 |
-
css=css
|
|
|
39 |
).launch(debug=True)
|
|
|
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')
|
|
|
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, 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']]
|
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,
|
37 |
examples=examples,
|
38 |
+
css=css,
|
39 |
+
enable_queue=True
|
40 |
).launch(debug=True)
|