lydianish commited on
Commit
03a9d25
1 Parent(s): d1ef2c2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -11,13 +11,13 @@ laser_vocab = f"{os.environ['LASER']}/models/laser2.cvocab"
11
  laser_tokenizer = 'spm'
12
  laser_model = RoLaserEncoder(model_path=laser_checkpoint, vocab=laser_vocab, tokenizer=laser_tokenizer)
13
 
14
- rolaser_checkpoint = f"{os.environ['ROLASER']}/models/RoLASER/rolaser.pt"
15
- rolaser_vocab = f"{os.environ['ROLASER']}/models/RoLASER/rolaser.cvocab"
16
  rolaser_tokenizer = 'roberta'
17
  rolaser_model = RoLaserEncoder(model_path=rolaser_checkpoint, vocab=rolaser_vocab, tokenizer=rolaser_tokenizer)
18
 
19
- c_rolaser_checkpoint = f"{os.environ['ROLASER']}/models/c-RoLASER/c-rolaser.pt"
20
- c_rolaser_vocab = f"{os.environ['ROLASER']}/models/c-RoLASER/c-rolaser.cvocab"
21
  c_rolaser_tokenizer = 'char'
22
  c_rolaser_model = RoLaserEncoder(model_path=c_rolaser_checkpoint, vocab=c_rolaser_vocab, tokenizer=c_rolaser_tokenizer)
23
 
 
11
  laser_tokenizer = 'spm'
12
  laser_model = RoLaserEncoder(model_path=laser_checkpoint, vocab=laser_vocab, tokenizer=laser_tokenizer)
13
 
14
+ rolaser_checkpoint = f"{os.environ['ROLASER']}/models/rolaser.pt"
15
+ rolaser_vocab = f"{os.environ['ROLASER']}/models/rolaser.cvocab"
16
  rolaser_tokenizer = 'roberta'
17
  rolaser_model = RoLaserEncoder(model_path=rolaser_checkpoint, vocab=rolaser_vocab, tokenizer=rolaser_tokenizer)
18
 
19
+ c_rolaser_checkpoint = f"{os.environ['ROLASER']}/models/c-rolaser.pt"
20
+ c_rolaser_vocab = f"{os.environ['ROLASER']}/models/c-rolaser.cvocab"
21
  c_rolaser_tokenizer = 'char'
22
  c_rolaser_model = RoLaserEncoder(model_path=c_rolaser_checkpoint, vocab=c_rolaser_vocab, tokenizer=c_rolaser_tokenizer)
23