Hugging Face's logo

Spaces:
AbstractPhil
/
bert-beatrix-2048-testing
Running on Zero

App Files Community
AbstractPhil commited on
Commit
bb1447d
·
verified ·
1 Parent(s): 47abe17

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -24,7 +24,7 @@ class SymbolicConfig:
24
  config = SymbolicConfig()
25
  model_dir = snapshot_download(repo_id=config.repo_id)
26
  tokenizer = AutoTokenizer.from_pretrained(model_dir)
27
- model = AutoModelForMaskedLM.from_pretrained(model_dir).eval().cuda()
28
 
29
 
30
  MASK_TOKEN = tokenizer.mask_token or "[MASK]"
 
24
  config = SymbolicConfig()
25
  model_dir = snapshot_download(repo_id=config.repo_id)
26
  tokenizer = AutoTokenizer.from_pretrained(model_dir)
27
+ model = AutoModelForMaskedLM.from_pretrained(model_dir, trust_remote_code=True).eval().cuda()
28
 
29
 
30
  MASK_TOKEN = tokenizer.mask_token or "[MASK]"