Update app.py

app.py

@@ -1,23 +1,37 @@
-import tempfile
 import os
 import streamlit as st
 import paddlehub as hub
 from PIL import Image
 import io
 
-pp_ocrv3 = hub.Module(name="ch_pp-ocrv3")
 
-def inference(img):
-  with tempfile.TemporaryDirectory() as tempdir_name:
-    pp_ocrv3.recognize_text(paths=[img],use_gpu=False,output_dir=tempdir_name,visualization=True)
-    result_names = os.listdir(tempdir_name)
-    output_image = Image.open(os.path.join(tempdir_name, result_names[0]))
-    return [output_image]
-  
-title="ch_PP-OCRv3"
-description="ch_PP-OCRv3 is a practical ultra-lightweight OCR system developed by PaddleOCR."
+from paddleocr import PaddleOCR,draw_ocr
+def ocr(img,lang):
+    # Paddleocr supports Chinese, English, French, German, Korean and Japanese.
+    # You can set the parameter `lang` as `ch`, `en`, `french`, `german`, `korean`, `japan`
+    # to switch the language model in order.
+    ocr = PaddleOCR(use_angle_cls=True, lang=lang) # need to run only once to download and load model into memory
+    img_path = 'PaddleOCR/doc/imgs_en/img_12.jpg'
+    result = ocr.ocr(img, cls=True)
+    for idx in range(len(result)):
+        res = result[idx]
+        for line in res:
+            print(line)
 
-examples=[['test.png']]
+    # draw result
+    from PIL import Image
+    result = result[0]
+    image = Image.open(img_path).convert('RGB')
+    boxes = [line[0] for line in result]
+    st.text(boxes)
+    txts = [line[1][0] for line in result]
+    st.text(txts)
+    scores = [line[1][1] for line in result]
+    st.text(scores)
+    #im_show = draw_ocr(image, boxes, txts, scores, font_path='/path/to/PaddleOCR/doc/fonts/simfang.ttf')
+    #im_show = Image.fromarray(im_show)
+    #st.image(im_show, caption='Sunrise by the mountains')
+    #im_show.save('result.jpg')
 
 
 uploaded_file = st.file_uploader("Choose a file")
@@ -26,4 +40,5 @@ if uploaded_file is not None:
     bytes_data = uploaded_file.getvalue()
     st.write(bytes_data)
     image = Image.open(io.BytesIO(bytes_data))
-    st.image(image, caption='Sunrise by the mountains')
+    st.image(image, caption='Sunrise by the mountains')
+    ocr(image,'ch')