Hugging Face's logo

Spaces:
BigSalmon
/
MaskSeveralAtOnce
Runtime error

App Files Community
BigSalmon commited on
Commit
f5e8576
•
1 Parent(s): b595091

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -12,7 +12,8 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
12
 
13
  from transformers import AutoTokenizer, AutoModelForMaskedLM
14
  tokenizer = AutoTokenizer.from_pretrained("roberta-large")
15
- model = AutoModelForMaskedLM.from_pretrained("BigSalmon/MrLincolnBerta")
 
16
  model2 = AutoModelForMaskedLM.from_pretrained("roberta-base")
17
 
18
 
 
12
 
13
  from transformers import AutoTokenizer, AutoModelForMaskedLM
14
  tokenizer = AutoTokenizer.from_pretrained("roberta-large")
15
+ model = AutoModelForMaskedLM.from_pretrained("BigSalmon/FormalRobertaLincoln")
16
+ #model = AutoModelForMaskedLM.from_pretrained("BigSalmon/MrLincolnBerta")
17
  model2 = AutoModelForMaskedLM.from_pretrained("roberta-base")
18
 
19