sudacode/owocr
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Properly chmod helper executable

Browse Source
This commit is contained in:
AuroraWright
2024-02-11 13:24:17 +01:00
parent 6c9ec56329
commit 4b998d24c3
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
owocr/ocr.py
View File

@@ -306,12 +306,15 @@ class AppleLiveText:
else: else:
self.helper_executable = os.path.join(os.path.expanduser('~'),'.cache','livetexthelper') self.helper_executable = os.path.join(os.path.expanduser('~'),'.cache','livetexthelper')
if not os.path.isfile(self.helper_executable): if not os.path.isfile(self.helper_executable):
logger.info('Downloading helper executable') logger.info('Downloading helper executable ' + self.helper_executable)
try: try:
cache_folder = os.path.join(os.path.expanduser('~'),'.cache') cache_folder = os.path.join(os.path.expanduser('~'),'.cache')
if not os.path.isdir(cache_folder): if not os.path.isdir(cache_folder):
os.makedirs(cache_folder) os.makedirs(cache_folder)
urllib.request.urlretrieve('https://github.com/AuroraWright/owocr/raw/master/livetexthelper', self.helper_executable) urllib.request.urlretrieve('https://github.com/AuroraWright/owocr/raw/master/livetexthelper', self.helper_executable)
mode = os.stat(self.helper_executable).st_mode
mode |= (mode & 0o444) >> 2
os.chmod(self.helper_executable, mode)
except: except:
logger.warning('Download failed. Apple Live Text will not work!') logger.warning('Download failed. Apple Live Text will not work!')
return return
@@ -493,7 +496,7 @@ class RapidOCR:
else: else:
rapidocr_model_file = os.path.join(os.path.expanduser('~'),'.cache','rapidocr_japan_PP-OCRv4_rec_infer.onnx') rapidocr_model_file = os.path.join(os.path.expanduser('~'),'.cache','rapidocr_japan_PP-OCRv4_rec_infer.onnx')
if not os.path.isfile(rapidocr_model_file): if not os.path.isfile(rapidocr_model_file):
logger.info('Downloading RapidOCR model') logger.info('Downloading RapidOCR model ' + rapidocr_model_file)
try: try:
cache_folder = os.path.join(os.path.expanduser('~'),'.cache') cache_folder = os.path.join(os.path.expanduser('~'),'.cache')
if not os.path.isdir(cache_folder): if not os.path.isdir(cache_folder):