PP-OCRv3-ch2 / app.py
ulysses115's picture
Duplicate from ulysses115/PP-OCRv3-ch
eea0bcd
import tempfile
import os
import gradio as gr
import paddlehub as hub
from PIL import Image
pp_ocrv3 = hub.Module(name="ch_pp-ocrv3")
def inference(img):
"""
[{'save_path': '', 'data': [{'text': '飞桨“大航海”计划', 'confidence': 0.9351316094398499, 'text_box_position': [[246, 16], [384, 14], [384, 33], [246, 35]]}, {'text': '产业智船化升圾', 'confidence': 0.8330735564231873, 'text_box_position': [[433, 109], [487, 109], [487, 119], [433, 119]]}]}]
"""
results = pp_ocrv3.recognize_text(paths=[img],use_gpu=False,visualization=False)
print(results)
res = []
data = results[0]['data']
for item in data:
if item['confidence'] > 0.75 and len(item['text']) > 0:
res.append(item['text'])
return res
title="ch_PP-OCRv3"
description="ch_PP-OCRv3 is a practical ultra-lightweight OCR system developed by PaddleOCR."
examples=[['test.png']]
gr.Interface(inference,gr.inputs.Image(type="filepath"),
outputs=[gr.Textbox(label="Output Message")],
title=title,
description=description,
examples=examples).launch(enable_queue=True)