diff --git a/photo_mask.py b/photo_mask.py index e28f68e..f7f936d 100644 --- a/photo_mask.py +++ b/photo_mask.py @@ -20,7 +20,7 @@ from config.ucloud import BUCKET from models import ZxPhrec, ZxPhhd from ucloud import ucloud -OCR = PaddleOCR(use_angle_cls=False, lang="ch", show_log=False) +OCR = PaddleOCR(use_angle_cls=False, lang="ch", show_log=False, gpu_id=1) def open_image(img_path):