Fix MLM translation

apps/mlm.py

@@ -131,7 +131,7 @@ def app(state):
         scores = predict(transformed_image, dict(caption_inputs))
     scores = softmax(scores)
     labels, values = get_top_5_predictions(scores)
-    filled_sentence = mlm_state.caption.replace("[MASK]", labels[0])
+    filled_sentence = mlm_state.caption.replace("[MASK]", labels[-1])
     st.write("**Filled Sentence**: " + filled_sentence)
     st.write( f"""**English Translation**: {translate(filled_sentence, 'en')}""")
     # newer_col1, newer_col2 = st.beta_columns([6,4])