Cicciokr commited on
Commit
1deb871
·
verified ·
1 Parent(s): 0659652

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -13,9 +13,9 @@ st.write("Inserisci un testo con il token [MASK] per vedere le previsioni del mo
13
  input_text = st.text_input("Testo:", value="Lorem ipsum dolor sit amet, [MASK] adipiscing elit.")
14
 
15
  modelname = "./models/latin_bert/"
16
- model = AutoModelForMaskedLM.from_pretrained(modelname)
17
  # Creare un pipeline di riempimento maschere
18
- tokenizer = AutoTokenizer.from_pretrained(modelname)
19
 
20
  fill_mask = pipeline("fill-mask", model=model, tokenizer=tokenizer)
21
 
 
13
  input_text = st.text_input("Testo:", value="Lorem ipsum dolor sit amet, [MASK] adipiscing elit.")
14
 
15
  modelname = "./models/latin_bert/"
16
+ model = AutoModelForMaskedLM.from_pretrained(modelname, local_files_only=True)
17
  # Creare un pipeline di riempimento maschere
18
+ tokenizer = AutoTokenizer.from_pretrained("./models/subword_tokenizer_latin/", local_files_only=True)
19
 
20
  fill_mask = pipeline("fill-mask", model=model, tokenizer=tokenizer)
21