Spaces:

BigSalmon
/

MaskSeveralAtOnce

Runtime error

BigSalmon commited on Dec 2, 2021

Commit

fb022f0

1 Parent(s): 2ac5309

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,7 +13,7 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 from transformers import AutoTokenizer, AutoModelForMaskedLM
 tokenizer = AutoTokenizer.from_pretrained("roberta-large")
-model = AutoModelForMaskedLM.from_pretrained("BigSalmon/FormalRobertaaa")
 model2 = AutoModelForMaskedLM.from_pretrained("roberta-base")

 from transformers import AutoTokenizer, AutoModelForMaskedLM
 tokenizer = AutoTokenizer.from_pretrained("roberta-large")
+model = AutoModelForMaskedLM.from_pretrained("BigSalmon/FormalRobertaa")
 model2 = AutoModelForMaskedLM.from_pretrained("roberta-base")