Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -28,15 +28,15 @@ input_text = st.text_input("Testo:", value="Lorem ipsum dolor sit amet, [MASK] a
|
|
28 |
modelname = "./models/bert-base-latin-uncased"
|
29 |
|
30 |
#ClassCat/roberta-base-latin-v2
|
31 |
-
tokenizer_robertaclasscat = AutoTokenizer.from_pretrained("
|
32 |
-
model_robertaclasscat = AutoModelForMaskedLM.from_pretrained("
|
33 |
fill_mask_robertaclasscat = pipeline("fill-mask", model=model_robertaclasscat, tokenizer=tokenizer_robertaclasscat)
|
34 |
|
35 |
|
36 |
|
37 |
|
38 |
-
tokenizer_robertapstroe = AutoTokenizer.from_pretrained("pstroe/roberta-base-latin-
|
39 |
-
model_robertapstroe = AutoModelForMaskedLM.from_pretrained("pstroe/roberta-base-latin-
|
40 |
fill_mask_robertapstroe = pipeline("fill-mask", model=model_robertapstroe, tokenizer=tokenizer_robertapstroe)
|
41 |
|
42 |
tokenizer = AutoTokenizer.from_pretrained(modelname)
|
|
|
28 |
modelname = "./models/bert-base-latin-uncased"
|
29 |
|
30 |
#ClassCat/roberta-base-latin-v2
|
31 |
+
tokenizer_robertaclasscat = AutoTokenizer.from_pretrained("Cicciokr/xlm-roberta-latin")
|
32 |
+
model_robertaclasscat = AutoModelForMaskedLM.from_pretrained("Cicciokr/xlm-roberta-latin")
|
33 |
fill_mask_robertaclasscat = pipeline("fill-mask", model=model_robertaclasscat, tokenizer=tokenizer_robertaclasscat)
|
34 |
|
35 |
|
36 |
|
37 |
|
38 |
+
tokenizer_robertapstroe = AutoTokenizer.from_pretrained("pstroe/roberta-base-latin-cased")
|
39 |
+
model_robertapstroe = AutoModelForMaskedLM.from_pretrained("pstroe/roberta-base-latin-cased")
|
40 |
fill_mask_robertapstroe = pipeline("fill-mask", model=model_robertapstroe, tokenizer=tokenizer_robertapstroe)
|
41 |
|
42 |
tokenizer = AutoTokenizer.from_pretrained(modelname)
|