joechou commited on
Commit
6a46fc7
·
1 Parent(s): 118ef37

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -33,8 +33,8 @@ def init():
33
  model = KobeModel(args)
34
 
35
 
36
- model = model.load_from_checkpoint("./epoch=19-step=66080.ckpt", args=args)
37
- #model = model.load_from_checkpoint("./epoch=11-step=396384.ckpt", args=args)
38
 
39
  text_tokenizer = BertTokenizer.from_pretrained(args.vocab_file)
40
  cond_tokenizer = spm.SentencePieceProcessor()
 
33
  model = KobeModel(args)
34
 
35
 
36
+ #model = model.load_from_checkpoint("./epoch=19-step=66080.ckpt", args=args)
37
+ model = model.load_from_checkpoint("./epoch=11-step=396384.ckpt", args=args)
38
 
39
  text_tokenizer = BertTokenizer.from_pretrained(args.vocab_file)
40
  cond_tokenizer = spm.SentencePieceProcessor()