Spaces:
Paused
Paused
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):
|