Update app.py

app.py

@@ -2,10 +2,10 @@
 
 #x = st.slider('Select a value')
 #st.write(x, 'squared is', x * x)
-from transformers import pipeline, BertForQuestionAnswering, AutoTokenizer
+from transformers import pipeline, AutoModelForMaskedLM, AutoTokenizer
 
-modelname = 'models/latin_bert'
-model = BertForQuestionAnswering.from_pretrained(modelname)
+modelname = "./models/latin_bert/"
+model = AutoModelForMaskedLM.from_pretrained(modelname)
 # Creare un pipeline di riempimento maschere
 tokenizer = AutoTokenizer.from_pretrained(modelname)