使用gpu1加载ocr模型

This commit is contained in:
2024-07-10 12:34:59 +08:00
parent 0a7e5b42e8
commit 534bf4b4b9

View File

@@ -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):