isLinXu commited on
Commit
7e870a5
·
1 Parent(s): c5fd475

update app

Browse files
Files changed (1) hide show
  1. app.py +41 -0
app.py ADDED
@@ -0,0 +1,41 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ '''
2
+ from modelscope.pipelines import pipeline
3
+ from modelscope.utils.constant import Tasks
4
+ import cv2
5
+
6
+ ocr_recognition = pipeline(Tasks.ocr_recognition, model='damo/cv_convnextTiny_ocr-recognition-general_damo')
7
+
8
+ ### 使用url
9
+ img_url = 'http://duguang-labelling.oss-cn-shanghai.aliyuncs.com/mass_img_tmp_20220922/ocr_recognition.jpg'
10
+ result = ocr_recognition(img_url)
11
+ print(result)
12
+ '''
13
+ import os
14
+ os.system("pip install tensorflow")
15
+ os.system("pip install modelscope")
16
+
17
+
18
+ import gradio as gr
19
+ import torch
20
+ from modelscope.pipelines import pipeline
21
+ from modelscope.utils.constant import Tasks
22
+
23
+ def ocr_recognition(img_url):
24
+ ocr_recognition = pipeline(Tasks.ocr_recognition, model='damo/cv_convnextTiny_ocr-recognition-general_damo')
25
+ result = ocr_recognition(img_url)
26
+ return result
27
+
28
+ def download_test_image():
29
+ # Images
30
+ torch.hub.download_url_to_file(
31
+ 'https://github.com/isLinXu/issues/assets/59380685/6cdbe53c-eb34-4310-8bd4-18b0a1aff803',
32
+ 'ocr_test.jpg')
33
+
34
+ download_test_image()
35
+ input_image = gr.inputs.Image()
36
+ output_text = gr.outputs.Textbox()
37
+ examples = [["ocr_test.jpg"]]
38
+
39
+ gr.Interface(fn=ocr_recognition, inputs=input_image,
40
+ outputs=output_text, examples=examples,
41
+ title="OCR Recognition").launch()