From 5f9bcd89f592fcd319b2408d03ddfa6b102e1133 Mon Sep 17 00:00:00 2001 From: AuroraWright Date: Tue, 30 Jan 2024 11:23:16 +0100 Subject: [PATCH] Fix module downloading --- owocr/ocr.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/owocr/ocr.py b/owocr/ocr.py index c10e424..8a22869 100644 --- a/owocr/ocr.py +++ b/owocr/ocr.py @@ -47,6 +47,7 @@ except ImportError: try: from rapidocr_onnxruntime import RapidOCR as ROCR + import urllib.request except ImportError: pass @@ -434,8 +435,8 @@ class RapidOCR: if not os.path.isfile(rapidocr_model_file): logger.info('Downloading RapidOCR model') try: - urllib.request.urlretrieve('https://raw.githubusercontent.com/AuroraWright/owocr/master/rapidocr_japan_PP-OCRv4_rec_infer.onnx', rapidocr_model_file) - except Exception as inst: + urllib.request.urlretrieve('https://github.com/AuroraWright/owocr/raw/master/rapidocr_japan_PP-OCRv4_rec_infer.onnx', rapidocr_model_file) + except: logger.warning('Download failed. RapidOCR will not work!') return