Ahsen Khaliq commited on
Commit
6f6e791
1 Parent(s): e67f683

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -5
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)