Fix put text bug

main.py

@@ -2,9 +2,8 @@ from mmocr.ocr import MMOCR
 import gradio as gr
 import cv2
 
-model_dir = 'model'
-ocr = MMOCR(det_config=f'{model_dir}/config.py',
-            det_ckpt=f'{model_dir}/epoch_40.pth', device='cpu')
+ocr = MMOCR(det_config='model/config.py',
+            det_ckpt='model/epoch_40.pth', device='cpu')
 def get_rec(points):
     xs = []
     ys = []
@@ -28,7 +27,7 @@ def predict(image_input, score_threshold):
         
         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)
+        draw_img = cv2.putText(draw_img, '%.3f' % score, p0, cv2.FONT_HERSHEY_PLAIN, 2, (0, 255, 0))
     return draw_img
 
 def run():