Update app.py

app.py

@@ -20,6 +20,11 @@ modelname_lv = "LuisAVasquez/simple-latin-bert-uncased"
 modelname = "./models/bert-base-latin-uncased"
 
 
+
+tokenizer_roberta = AutoTokenizer.from_pretrained("pstroe/roberta-base-latin-cased3")
+model_roberta = AutoModelForMaskedLM.from_pretrained("pstroe/roberta-base-latin-cased3")
+fill_mask_roberta = pipeline("fill-mask", model=model_roberta, tokenizer=tokenizer_roberta)
+
 tokenizer = AutoTokenizer.from_pretrained(modelname)
 model = AutoModelForMaskedLM.from_pretrained(modelname)
 fill_mask = pipeline("fill-mask", model=model, tokenizer=tokenizer)
@@ -37,5 +42,9 @@ if input_text:
     st.subheader("Risultati delle previsioni con Simple Latin Bert:")
     for pred_lv in predictions_lv:
         st.write(f"**Parola**: {pred_lv['token_str']}, **Probabilità**: {pred_lv['score']:.4f}, **Sequence**: {pred_lv['sequence']}")
+    predictions_roberta = fill_mask_roberta(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']}")