浏览代码

fix(magic_pdf): remove unused import from pdf_extract_kit

myhloli 1 年之前
父节点
当前提交
7ecc82da63
共有 1 个文件被更改,包括 1 次插入2 次删除
  1. 1 2
      magic_pdf/model/pdf_extract_kit.py

+ 1 - 2
magic_pdf/model/pdf_extract_kit.py

@@ -9,8 +9,7 @@ try:
     import argparse
     import numpy as np
     import torch
-
-    from paddleocr import draw_ocr
+    import torchtext;torchtext.disable_torchtext_deprecation_warning()
     from PIL import Image
     from torchvision import transforms
     from torch.utils.data import Dataset, DataLoader