Spaces:
Sleeping
Sleeping
Update document.py
Browse files- document.py +2 -2
document.py
CHANGED
@@ -128,8 +128,8 @@ class Document():
|
|
128 |
|
129 |
def merge(self, minPage, maxPage, model, device):
|
130 |
model.eval()
|
131 |
-
if
|
132 |
-
return len(self.segmentation['text']
|
133 |
|
134 |
lossScore = []
|
135 |
for i in trange(len(self.segmentation['text']) - 1):
|
|
|
128 |
|
129 |
def merge(self, minPage, maxPage, model, device):
|
130 |
model.eval()
|
131 |
+
if minPage > len(self.segmentation['text']):
|
132 |
+
return len(self.segmentation['text']
|
133 |
|
134 |
lossScore = []
|
135 |
for i in trange(len(self.segmentation['text']) - 1):
|