Fix rectangle color

main.py

@@ -18,10 +18,11 @@ def get_rec(points):
     
 def predict(image_input):
     draw_img = image_input.copy()
+    print(image_input.shape)
     output = ocr.readtext(image_input)
     for polygon in output['det_polygons']:
         p0, p1 = get_rec([int(i) for i in polygon])
-        draw_img = cv2.rectangle(draw_img, p0, p1, (255,255,255))
+        draw_img = cv2.rectangle(draw_img, p0, p1, (255,0,0), 2)
     return draw_img
 
 def run():