com3dian commited on
Commit
6a6b705
·
verified ·
1 Parent(s): f19afe4

Update document.py

Browse files
Files changed (1) hide show
  1. 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 maxPage < len(self.segmentation['text']):
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):