Update

main.py

@@ -1,6 +1,7 @@
 from mmocr.ocr import MMOCR
 import gradio as gr
 import cv2
+import sys
 
 ocr = MMOCR(det_config='model/config.py',
             det_ckpt='model/epoch_40.pth', device='cpu')
@@ -17,10 +18,13 @@ def get_rec(points):
     
 def predict(image_input, score_threshold):
     draw_img = image_input.copy()
+    print('image shape', image_input.shape, file=sys.stderr)
+
     try:
         output = ocr.readtext(image_input)
     except:
         output = dict(det_polygons=[], det_scores=[])
+
     polygons = output['det_polygons']
     scores = output['det_scores']
     for polygon, score in zip(polygons, scores):
@@ -28,6 +32,7 @@ def predict(image_input, score_threshold):
             continue
         p0, p1 = get_rec([int(i) for i in polygon])
         draw_img = cv2.rectangle(draw_img, p0, p1, (255,0,0), 2)
+
     return draw_img
 
 def run():