Remove eos in sentence

source/predict_sample.py

@@ -73,12 +73,12 @@ def generate_caption(image: torch.Tensor,
 
         # id --> word
         next_word_pred = vocab.index_to_word(int(next_id_pred.item()))
-        sentence.append(next_word_pred)
 
-        # stop if we predict '<eos>'
         if next_word_pred == vocab.index2word[vocab.EOS]:
             break
 
+        sentence.append(next_word_pred)
+
     return sentence