Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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(
|
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 |
|