Commit 46cacbc0 authored by myhloli's avatar myhloli

fix(magic_pdf): handle import errors with exception logging

parent 7952ee4a
from loguru import logger from loguru import logger
import os import os
import time
try: try:
import cv2 import cv2
import yaml import yaml
import time
import argparse import argparse
import numpy as np import numpy as np
import torch import torch
...@@ -20,7 +20,8 @@ try: ...@@ -20,7 +20,8 @@ try:
from magic_pdf.model.pek_sub_modules.layoutlmv3.model_init import Layoutlmv3_Predictor from magic_pdf.model.pek_sub_modules.layoutlmv3.model_init import Layoutlmv3_Predictor
from magic_pdf.model.pek_sub_modules.post_process import get_croped_image, latex_rm_whitespace from magic_pdf.model.pek_sub_modules.post_process import get_croped_image, latex_rm_whitespace
from magic_pdf.model.pek_sub_modules.self_modify import ModifiedPaddleOCR from magic_pdf.model.pek_sub_modules.self_modify import ModifiedPaddleOCR
except ImportError: except ImportError as e:
logger.exception(e)
logger.error('Required dependency not installed, please install by \n"pip install magic-pdf[full-cpu] detectron2 --extra-index-url https://myhloli.github.io/wheels/"') logger.error('Required dependency not installed, please install by \n"pip install magic-pdf[full-cpu] detectron2 --extra-index-url https://myhloli.github.io/wheels/"')
exit(1) exit(1)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment