init change latin bert

app.py

@@ -1,4 +1,19 @@
-import streamlit as st
+#import streamlit as st
 
-x = st.slider('Select a value')
-st.write(x, 'squared is', x * x)
+#x = st.slider('Select a value')
+#st.write(x, 'squared is', x * x)
+from transformers import pipeline, BertForQuestionAnswering, AutoTokenizer
+
+modelname = 'models/latin_bert'
+model = BertForQuestionAnswering.from_pretrained(modelname)
+# Creare un pipeline di riempimento maschere
+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:
+st.write(risultato["sequence"])