Update app.py

app.py

@@ -47,7 +47,8 @@ if input_text:
     st.subheader("Risultati delle previsioni con Simple Latin Bert:")
     for pred_lv in predictions_lv:
         st.write(f"**Parola**: {pred_lv['token_str']}, **Probabilità**: {pred_lv['score']:.4f}, **Sequence**: {pred_lv['sequence']}")
-    predictions_roberta = fill_mask_roberta(input_text)    
+    input_text_roberta = input_text.replace("[MASK]", "<mask>")
+    predictions_roberta = fill_mask_roberta(input_text_roberta)    
     st.subheader("Risultati delle previsioni con Roberta Base Latin Cased 3:")
     for pred_roberta in predictions_roberta:
         st.write(f"**Parola**: {pred_roberta['token_str']}, **Probabilità**: {pred_roberta['score']:.4f}, **Sequence**: {pred_roberta['sequence']}")