Spaces:

Cicciokr
/

AIGenMaskedFillLatinText

Sleeping

Cicciokr commited on Jan 14

Commit

777e4db

verified ·

1 Parent(s): 3f6eb21

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -27,9 +27,9 @@ input_text = st.text_input("Testo:", value="Lorem ipsum dolor sit amet, [MASK] a
 #https://github.com/dbamman/latin-bert
 modelname = "./models/bert-base-latin-uncased"
-tokenizer_robertaclasscat = AutoTokenizer.from_pretrained("ClassCat/roberta-base-latin-v2")
-model_robertaclasscat = AutoModelForMaskedLM.from_pretrained("ClassCat/roberta-base-latin-v2")
 fill_mask_robertaclasscat = pipeline("fill-mask", model=model_robertaclasscat, tokenizer=tokenizer_robertaclasscat)

 #https://github.com/dbamman/latin-bert
 modelname = "./models/bert-base-latin-uncased"
+#ClassCat/roberta-base-latin-v2
+tokenizer_robertaclasscat = AutoTokenizer.from_pretrained("Cicciokr/xlm-roberta-latin")
+model_robertaclasscat = AutoModelForMaskedLM.from_pretrained("Cicciokr/xlm-roberta-latin")
 fill_mask_robertaclasscat = pipeline("fill-mask", model=model_robertaclasscat, tokenizer=tokenizer_robertaclasscat)