en-zh short sentences error

app.py

@@ -150,12 +150,13 @@ def split_token_from_sequences(sequences, model) -> dict :
 				
 				# new_id = step_w +'-' + str(step_i)
 				# parent_id = words_by_step[step_i-1][i_bs] + '-' + str(step_i -1)
-				
+				next_word_flag = 1
 				if step_i == 0 : 
 					parent_id =  'bos--1'
 				## if the dict already exists remove it, if it is not a root... 
 				## root?? then next is ''
-				next_word_flag = len(gen_sequences_texts[i_bs][step_i]) > step_i 
+				else:
+					next_word_flag = len(gen_sequences_texts[i_bs][step_i]) > step_i ## Not in step_i = 0;
 				if next_word_flag:
 					if not (new_id in id_dict_pos):
 						dict_parent.append({'id': new_id, 'parentId': parent_id , 'text': step_w, 'name': step_w, 'prob' : score })