Add filled sentence to MLM

apps/mlm.py

@@ -130,7 +130,9 @@ def app(state):
         scores = predict(transformed_image, dict(caption_inputs))
     scores = softmax(scores)
     labels, values = get_top_5_predictions(scores)
-    print(labels)
+    filled_sentence = mlm_state.caption.replace("[MASK]", labels[0])
+    st.write("Filled Sentence: " + filled_sentence)
+    st.write( f"""**English Translation**: {translate(filled_sentence, 'en')}""")
     # newer_col1, newer_col2 = st.beta_columns([6,4])
     fig = plotly_express_horizontal_bar_plot(values, labels)
     st.dataframe(pd.DataFrame({"Tokens":labels, "English Translation": list(map(lambda x: translate(x),labels))}).T)