Update segmentation.py

segmentation.py

@@ -60,12 +60,10 @@ class SemanticTextSegmentation:
         segments = self._text_tilling()
         merge_index = self._merge_segments(segments, threshold)
         new_segments = []
-        i_arr = []
         for i in merge_index:
-            i_arr.append(i)
             seg = ' '.join([segments[_] for _ in i])
             new_segments.append(seg)
-        return new_segments, i_arr
+        return new_segments
 
     def _merge_segments(self, segments, threshold):
         segment_map = [0]