fix bugs: no table detected

jayllfpt · Dec 2, 2024 · e939e87 · e939e87
1 parent b5a6419
commit e939e87
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/setup.py b/setup.py
@@ -15,7 +15,7 @@ def load_requirements(filename='requirements.txt'):
 
 setup(
     name="table2html",
-    version="1.4.0",
+    version="1.4.1",
     author="TraiPPN",
     license='Apache License 2.0',
     author_email="phamphungoctraivl@gmail.com",

diff --git a/table2html/main.py b/table2html/main.py
@@ -32,12 +32,14 @@ def __init__(
 
     def TableDetect(self, image):
         tables = self.table_detector(image)
-        return [
-            {
+        if len(tables):
+            return [
+                {
                 "table_bbox": table,
             }
-            for table in tables
-        ]
+                for table in tables
+            ]
+        return []
 
     def StructureDetect(self, table_image):
         # Detect rows and columns