Remove text

main.py

@@ -17,17 +17,17 @@ def get_rec(points):
     
 def predict(image_input, score_threshold):
     draw_img = image_input.copy()
-    print(image_input.shape)
-    output = ocr.readtext(image_input)
+    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):
         if score < 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)
-        draw_img = cv2.putText(draw_img, '%.3f' % score, p0, cv2.FONT_HERSHEY_PLAIN, 2, (0, 255, 0))
     return draw_img
 
 def run():