Hugging Face's logo

Spaces:
BigSalmon
/
MaskSeveralAtOnce
Runtime error

App Files Community
Simon Salmon commited on
Commit
5f9a357
·
1 Parent(s): ffe12e1

Update app.py

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