Update app.py

app.py

@@ -66,7 +66,7 @@ def get_valid_predictions(sentence, max_attempts=3, top_k=5):
     print(sentence)
     print(sentence in data["phrases"])
     if sentence in data["phrases"]:
-        return [{"token_str": data["phrases"][sentence], "score": 1.0, "sequence": sentence.replace("[MASK]", data["phrases"][sentence])}]
+        return [{"token_str": data["phrases"][sentence], "score": 1.0, "sequence": sentence.replace("<mask>", data["phrases"][sentence])}]
     
     attempt = 0
     filtered_predictions = []
@@ -83,15 +83,15 @@ def get_valid_predictions(sentence, max_attempts=3, top_k=5):
 
 # Imposta una chiave di sessione per il testo input, così possiamo aggiornarlo
 if "input_text_value" not in st.session_state:
-    st.session_state["input_text_value"] = "Lorem ipsum dolor sit amet, [MASK] adipiscing elit."
+    st.session_state["input_text_value"] = "Lorem ipsum dolor sit amet, <mask> adipiscing elit."
 
 # Frasi di esempio
 examples = [
-    "Asdrubal, frater Annibalis, qui secundo Punico bello [MASK] ingentibus copiis ab Hispania veniens ...",
-    "hanno et mago qui [MASK] punico bello cornelium consulem aput liparas ceperunt ...",
-    "Lorem ipsum dolor sit amet, [MASK] adipiscing elit.",
-    "Populus Romanus cum Macedonibus [MASK] ter gessit",
-    "Reliqui qui tum principes numerabantur in magistratibus [MASK] cotidieque fere a nobis in contionibus audiebantur."
+    "Asdrubal, frater Annibalis, qui secundo Punico bello <mask> ingentibus copiis ab Hispania veniens ...",
+    "hanno et mago qui <mask> punico bello cornelium consulem aput liparas ceperunt ...",
+    "Lorem ipsum dolor sit amet, <mask> adipiscing elit.",
+    "Populus Romanus cum Macedonibus <mask> ter gessit",
+    "Reliqui qui tum principes numerabantur in magistratibus <mask> cotidieque fere a nobis in contionibus audiebantur."
 ]
 
 st.title("Completamento di parole in testi Latino Antico con Analisi Morfologica")
@@ -140,9 +140,6 @@ try:
 except ImportError:
     nlp_lat = None
 
-if "input_text_value" not in st.session_state:
-    st.session_state["input_text_value"] = "Lorem ipsum dolor sit amet, [MASK] adipiscing elit."
-
 tokenizer_roberta = AutoTokenizer.from_pretrained("Cicciokr/Roberta-Base-Latin-Uncased")
 model_roberta = AutoModelForMaskedLM.from_pretrained("Cicciokr/Roberta-Base-Latin-Uncased")
 fill_mask_roberta = pipeline("fill-mask", model=model_roberta, tokenizer=tokenizer_roberta)