Update app.py

app.py

@@ -11,10 +11,13 @@ st.write("Inserisci un testo con il token [MASK] per vedere le previsioni del mo
 
 
 st.write("Esempi di testo:");
-st.write("duces et reges carthaginiensivm hanno et mago qui [MASK] punico bello cornelium consulem aput liparas ceperunt");
-st.write("hanno et mago qui [MASK]  punico bello cornelium consulem aput liparas ceperunt");
-#duces et reges carthaginiensivm hanno et mago qui [MASK] punico bello cornelium consulem aput liparas ceperunt
-#hanno et mago qui [MASK]  punico bello cornelium consulem aput liparas ceperunt
+st.write("Asdrubal, frater Annibalis, qui secundo Punico bello [MASK] ingentibus copiis ab Hispania veniens => cum");
+st.write("hanno et mago qui [MASK]  punico bello cornelium consulem aput liparas ceperunt => primo");
+st.write("Lorem ipsum dolor sit amet, [MASK] adipiscing elit. => consectetur");
+st.write("Populus Romanus cum Macedonibus [MASK] ter gessit => bellum");
+#Asdrubal, frater Annibalis, qui secundo Punico bello [MASK] ingentibus copiis ab Hispania veniens => cum
+#hanno et mago qui [MASK]  punico bello cornelium consulem aput liparas ceperunt => primo
+#Lorem ipsum dolor sit amet, [MASK] adipiscing elit. => consectetur
 input_text = st.text_input("Testo:", value="Lorem ipsum dolor sit amet, [MASK] adipiscing elit.")
 
 # Model based on BERT
@@ -30,6 +33,10 @@ tokenizer_roberta = AutoTokenizer.from_pretrained("pstroe/roberta-base-latin-cas
 model_roberta = AutoModelForMaskedLM.from_pretrained("pstroe/roberta-base-latin-cased3")
 fill_mask_roberta = pipeline("fill-mask", model=model_roberta, tokenizer=tokenizer_roberta)
 
+tokenizer_robertaclasscat = AutoTokenizer.from_pretrained("ClassCat/roberta-base-latin-v2")
+model_robertaclasscat = AutoModelForMaskedLM.from_pretrained("ClassCat/roberta-base-latin-v2")
+fill_mask_robertaclasscat = pipeline("fill-mask", model=model_robertaclasscat, tokenizer=tokenizer_robertaclasscat)
+
 tokenizer = AutoTokenizer.from_pretrained(modelname)
 model = AutoModelForMaskedLM.from_pretrained(modelname)
 fill_mask = pipeline("fill-mask", model=model, tokenizer=tokenizer)
@@ -52,5 +59,9 @@ if input_text:
     st.subheader("Risultati delle previsioni con Roberta Base Latin Cased 3:")
     for pred_roberta in predictions_roberta:
         st.write(f"**Parola**: {pred_roberta['token_str']}, **Probabilità**: {pred_roberta['score']:.4f}, **Sequence**: {pred_roberta['sequence']}")
+    predictions_robertaclasscat = fill_mask_robertaclasscat(input_text)    
+    st.subheader("Risultati delle previsioni con Roberta Base Latin ClassCat V2:")
+    for pred_robertaclasscat in predictions_robertaclasscat:
+        st.write(f"**Parola**: {pred_robertaclasscat['token_str']}, **Probabilità**: {pred_robertaclasscat['score']:.4f}, **Sequence**: {pred_robertaclasscat['sequence']}")