rodrigomasini commited on
Commit
75717e3
·
verified ·
1 Parent(s): 7038aec

Update mdr_pdf_parser.py

Browse files
Files changed (1) hide show
  1. 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=="cpu"),
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"],