Commit dba28389 authored by myhloli's avatar myhloli
Browse files

refactor(model): update model downloads and disable unused models

- Comment out LayoutLMv3, TableMaster, and StructEqTable models
- Update MFR model path to unimernet_hf_small_2503- Remove unused import in Unimernet.py
parent 2f40fa7d
...@@ -30,7 +30,6 @@ class UnimernetModel(object): ...@@ -30,7 +30,6 @@ class UnimernetModel(object):
self.model = self.model.to(dtype=torch.float16) self.model = self.model.to(dtype=torch.float16)
self.model.eval() self.model.eval()
def predict(self, mfd_res, image): def predict(self, mfd_res, image):
formula_list = [] formula_list = []
mf_image_list = [] mf_image_list = []
......
...@@ -4,12 +4,12 @@ from huggingface_hub import snapshot_download ...@@ -4,12 +4,12 @@ from huggingface_hub import snapshot_download
if __name__ == "__main__": if __name__ == "__main__":
mineru_patterns = [ mineru_patterns = [
"models/Layout/LayoutLMv3/*", # "models/Layout/LayoutLMv3/*",
"models/Layout/YOLO/*", "models/Layout/YOLO/*",
"models/MFD/YOLO/*", "models/MFD/YOLO/*",
"models/MFR/unimernet_small_2501/*", "models/MFR/unimernet_hf_small_2503/*",
"models/TabRec/TableMaster/*", # "models/TabRec/TableMaster/*",
"models/TabRec/StructEqTable/*", # "models/TabRec/StructEqTable/*",
] ]
model_dir = snapshot_download( model_dir = snapshot_download(
"opendatalab/PDF-Extract-Kit-1.0", "opendatalab/PDF-Extract-Kit-1.0",
......
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