Update app.py

app.py

@@ -34,7 +34,7 @@ if st.button("Submit"):
 
     user_predictions = unmasker(user_masked_sentence)
     
-    st.write(user_predictions)  
+    # st.write(user_predictions)  
 
     if len(user_predictions) > 0:
         # st.write(f"Top prediction for the masked token: {user_predictions[0]['sequence']}")
@@ -44,7 +44,8 @@ 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]['sequence'])  
+            predicted_sentence = replace_mask(original_sentence, language_predictions[0]['token_str'])  
+            st.write(language_predictions)
             st.write(f"Original sentence ({language}): {original_sentence}")
             st.write(f"Top prediction for the masked token: {predicted_sentence}\n")
             st.write("=" * 80)