wip

app.py

@@ -1,6 +1,8 @@
 import streamlit as st
+import json
 import os
 import cv2
+from utils import draw_ocr_results, start_ocr
 
 def get_image_path(img):
     # Create a directory and save the uploaded image.
@@ -15,7 +17,24 @@ if uploaded_file is not None:
     # Get actual image file
     fpath= get_image_path(uploaded_file)
     st.image(fpath)
-    item = cv2.imread(fpath)
-    item = cv2.cvtColor(item, cv2.COLOR_BGR2GRAY)
+    res = start_ocr({
+    'image_dir': fpath,
+    'drop_score': 0.3,
+    #'det_model_dir': "./paddle/ch_PP-OCRv3_det_infer",
+    'det_model_dir': "./paddle/ch_PP-OCRv4_det_infer",
+    #'det_model_dir': "./paddle/ch_PP-OCRv4_det_server_infer",
 
-    st.image(item)
+    #'rec_model_dir': "./paddle/ch_PP-OCRv3_rec_infer",
+    'rec_model_dir': "./paddle/ch_PP-OCRv4_rec_infer",
+    #'rec_model_dir': "./paddle/ch_PP-OCRv4_rec_server_infer",
+
+    'det_limit_side_len': 2496,
+    "det_db_box_thresh": 0.2,
+
+    "use_angle_cls": False,
+    #"cls_model_dir": "./paddle/ch_ppocr_mobile_v2.0_cls_infer",
+    "rec_char_dict_path": './paddle/ppocr_keys_v1.txt',
+    "show_log": True,
+})
+
+    st.write(json.dumps(res))