Cicciokr commited on
Commit
64c52ec
·
verified ·
1 Parent(s): ceb1520

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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("ClassCat/roberta-base-latin-v2")
32
- model_robertaclasscat = AutoModelForMaskedLM.from_pretrained("ClassCat/roberta-base-latin-v2")
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-cased3")
39
- model_robertapstroe = AutoModelForMaskedLM.from_pretrained("pstroe/roberta-base-latin-cased3")
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)