anh-khoa-nguyen commited on
Commit
c931053
·
1 Parent(s): 0338894

Specify writable cache directory for PaddleOCR models

Browse files
Files changed (2) hide show
  1. .gitignore +1 -0
  2. core/extractor.py +1 -1
.gitignore ADDED
@@ -0,0 +1 @@
 
 
1
+ /.paddleocr/
core/extractor.py CHANGED
@@ -27,7 +27,7 @@ class Extractor:
27
  self.config['device'] = 'cpu'
28
 
29
  if (ocr == None):
30
- self.ocr = PaddleOCR(lang='en')
31
  else:
32
  self.ocr = ocr
33
  if (detector == None):
 
27
  self.config['device'] = 'cpu'
28
 
29
  if (ocr == None):
30
+ self.ocr = PaddleOCR(lang='en', use_gpu=False, ocr_version='PP-OCRv3', det_model_dir='./.paddleocr/det', rec_model_dir='./.paddleocr/rec', cls_model_dir='./.paddleocr/cls')
31
  else:
32
  self.ocr = ocr
33
  if (detector == None):