fixed omission of end_indices in decode_batch return

Author: bsh98

model_generator.py

@@ -132,7 +132,8 @@ def decode_batch(self, bodies, special_append=None, max_output_length=100, sampl
         elif return_scores:
             return outputs, scores.tolist()
         else:
-            return outputs
+            # return outputs
+            return outputs, end_indices
 
     def decode_beam_batch(self, bodies, beam_size=3, max_output_length=100, sample=False):
         if self.mode != 'eval':