Update mdr_pdf_parser.py
Browse files- mdr_pdf_parser.py +1 -1
mdr_pdf_parser.py
CHANGED
@@ -1630,7 +1630,7 @@ class MDROcrEngine:
|
|
1630 |
paths = {k: str(Path(self._model_dir)/Path(*p)) for k,p in _MDR_OCR_MODELS.items()}
|
1631 |
# In MDROcrEngine._get_system()
|
1632 |
self._onnx_params = _MDR_ONNXParams(
|
1633 |
-
use_gpu=(self._device=="
|
1634 |
det_model_dir=paths["det"],
|
1635 |
cls_model_dir=paths["cls"],
|
1636 |
rec_model_dir=paths["rec"],
|
|
|
1630 |
paths = {k: str(Path(self._model_dir)/Path(*p)) for k,p in _MDR_OCR_MODELS.items()}
|
1631 |
# In MDROcrEngine._get_system()
|
1632 |
self._onnx_params = _MDR_ONNXParams(
|
1633 |
+
use_gpu=(self._device=="cuda"),
|
1634 |
det_model_dir=paths["det"],
|
1635 |
cls_model_dir=paths["cls"],
|
1636 |
rec_model_dir=paths["rec"],
|