Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 |
-
|
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()
|