joechou commited on
Commit
6a5d569
·
1 Parent(s): d7cf0ef

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -31,7 +31,7 @@ args.vocab_file= "bert-base-chinese"
31
  args.cond_vocab_file = "./vocab.cond.model"
32
  add_args(args)
33
 
34
- '''
35
 
36
  #model = KobeModel(args)
37
 
@@ -56,11 +56,12 @@ cond_tokenizer.Load(args.cond_vocab_file)
56
 
57
 
58
 
 
59
  #model = model.load_from_checkpoint("/root/kobe-v2/1ja19m5t/checkpoints/epoch=19-step=66080.ckpt", args=args)
60
  #model = model.load_from_checkpoint("/root/kobe-v2/37ht1cvz/checkpoints/epoch=11-step=396384.ckpt", args=args)
61
 
62
  trainer = pl.Trainer(accelerator='gpu', devices=1, max_epochs=-1)
63
-
64
 
65
 
66
  import streamlit as st
 
31
  args.cond_vocab_file = "./vocab.cond.model"
32
  add_args(args)
33
 
34
+
35
 
36
  #model = KobeModel(args)
37
 
 
56
 
57
 
58
 
59
+
60
  #model = model.load_from_checkpoint("/root/kobe-v2/1ja19m5t/checkpoints/epoch=19-step=66080.ckpt", args=args)
61
  #model = model.load_from_checkpoint("/root/kobe-v2/37ht1cvz/checkpoints/epoch=11-step=396384.ckpt", args=args)
62
 
63
  trainer = pl.Trainer(accelerator='gpu', devices=1, max_epochs=-1)
64
+ '''
65
 
66
 
67
  import streamlit as st