Spaces:
Paused
Paused
Update document.py
Browse files- document.py +1 -1
document.py
CHANGED
|
@@ -128,7 +128,7 @@ 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 = []
|
|
|
|
| 128 |
|
| 129 |
def merge(self, minPage, maxPage, model, device):
|
| 130 |
model.eval()
|
| 131 |
+
if maxPage < len(self.segmentation['text']):
|
| 132 |
return len(self.segmentation['text'])
|
| 133 |
|
| 134 |
lossScore = []
|