Update app.py

app.py

@@ -2,8 +2,16 @@
 
 #x = st.slider('Select a value')
 #st.write(x, 'squared is', x * x)
+import streamlit as st
 from transformers import pipeline, AutoModelForMaskedLM, AutoTokenizer
 
+
+st.title("Completamento del testo in Latino con Latin BERT")
+st.write("Inserisci un testo con il token [MASK] per vedere le previsioni del modello.")
+
+
+input_text = st.text_input("Testo:", value="Lorem ipsum dolor sit amet, [MASK] adipiscing elit.")
+
 modelname = "./models/latin_bert/"
 model = AutoModelForMaskedLM.from_pretrained(modelname)
 # Creare un pipeline di riempimento maschere
@@ -11,9 +19,8 @@ tokenizer = AutoTokenizer.from_pretrained(modelname)
 
 fill_mask = pipeline("fill-mask", model=model, tokenizer=tokenizer)
 
-
-# Testare il modello
-frase = "Gallia est omnis divisa in [MASK] tres."
-output = fill_mask(frase)
-for risultato in output:
-    print(risultato["sequence"])
+if input_text:
+    predictions = fill_mask(input_text)
+    st.subheader("Risultati delle previsioni:")
+    for pred in predictions:
+        st.write(f"**Parola**: {pred['token_str']}, **Probabilità**: {pred['score']:.4f}")