change table unitest dir

opendatalab · Nov 11, 2024 · 78b0268 · 78b0268
1 parent 4c9c555
commit 78b0268
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/unittest/test_table/test_tablemaster.py b/tests/unittest/test_table/test_tablemaster.py
@@ -6,10 +6,10 @@
 
 class TestppTableModel(unittest.TestCase):
     def test_image2html(self):
-        img = Image.open("tests/test_table/assets/table.jpg")
+        img = Image.open("tests/unittest/test_table/assets/table.jpg")
         # 修改table模型路径
         config = {"device": "cuda",
-                  "model_dir": "D:/models/PDF-Extract-Kit/models/TabRec/TableMaster"}
+                  "model_dir": "/home/quyuan/.cache/modelscope/hub/opendatalab/PDF-Extract-Kit/models/TabRec/TableMaster"}
         table_model = ppTableModel(config)
         res = table_model.img2html(img)
         # 验证生成的 HTML 是否符合预期