Update app.py

app.py

@@ -32,3 +32,10 @@ if input_text:
     result = unmasker(input_text)
     for pred_due in result:
         st.write(f"**Parola**: {pred_due['token_str']}, **Probabilità**: {pred_due['score']:.4f}, **Sequence**: {pred_due['sequence']}")
+
+    st.subheader("Risultati delle previsioni con bert-base-latin-uncased:")
+    path_to_bert_base = "./models/bert-base-latin-uncased/"
+    unmasker_bert_base = pipeline('fill-mask', model=path_to_bert_base)
+    result_bert_base = unmasker_bert_base(input_text)
+    for pred_tre in result_bert_base:
+        st.write(f"**Parola**: {pred_tre['token_str']}, **Probabilità**: {pred_tre['score']:.4f}, **Sequence**: {pred_tre['sequence']}")