Cicciokr commited on
Commit
6543fb9
·
verified ·
1 Parent(s): 777e4db

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -28,8 +28,8 @@ 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("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
 
 
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