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

Update document.py

Browse files
Files changed (1) hide show
  1. document.py +1 -1
document.py CHANGED
@@ -129,7 +129,7 @@ class Document():
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):
 
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):