Update app.py

app.py

@@ -44,7 +44,7 @@ if st.button("Submit"):
         
         for language, language_predictions in predictions.items():
             original_sentence = sample_sentences[language]
-            predicted_sentence = replace_mask(original_sentence, language_predictions[0]['token_str'])  
+            predicted_sentence = replace_mask(original_sentence, language_predictions[0]['sequence'])  
             st.write(f"Original sentence ({language}): {original_sentence}")
             st.write(f"Top prediction for the masked token: {predicted_sentence}\n")
             st.write("=" * 80)