Update app.py

app.py

@@ -22,6 +22,14 @@ fill_mask = pipeline("fill-mask", model=model, tokenizer=tokenizer)
 
 if input_text:
     predictions = fill_mask(input_text)
-    st.subheader("Risultati delle previsioni:")
+    st.subheader("Risultati delle previsioni con Simple Latin Bert:")
     for pred in predictions:
         st.write(f"**Parola**: {pred['token_str']}, **Probabilità**: {pred['score']:.4f}")
+
+
+fill_mask_due = pipeline("fill-mask", model="./models/latin_bert/")
+if input_text:
+    predictions_due = fill_mask_due(input_text)
+    st.subheader("Risultati delle previsioni con Latin Bert:")
+    for pred in predictions_due:
+        st.write(f"**Parola**: {pred['token_str']}, **Probabilità**: {pred['score']:.4f}")