Update app.py

app.py

@@ -26,10 +26,8 @@ if input_text:
     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}")
+    path_to_latin_bert = "./models/latin_bert/"
+    unmasker = pipeline('fill-mask', model=path_to_latin_bert)
+    result = unmasker("""dvces et reges carthaginiensivm hanno et mago qui [MASK] punico bello cornelium consulem aput liparas ceperunt""")
+    print(result)