wasmdashai commited on
Commit
a930a0e
·
verified ·
1 Parent(s): 7f8e3fa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -595,7 +595,7 @@ pro=TrinerModelVITS(dir_model=dir_model,
595
  @spaces.GPU
596
  def run_train_epoch(num):
597
  pro.init_training()
598
- for i in range(10):
599
  # model.train(True)
600
  yield pro.run_train_epoch()
601
 
@@ -619,7 +619,8 @@ def init_wandb():
619
  @spaces.GPU
620
  def save_pretrained(path):
621
  pro.save_pretrained(path)
622
- pro.init_training()
 
623
  return 'save_pretrained'
624
 
625
  with gr.Blocks() as interface:
 
595
  @spaces.GPU
596
  def run_train_epoch(num):
597
  pro.init_training()
598
+ for i in range(num):
599
  # model.train(True)
600
  yield pro.run_train_epoch()
601
 
 
619
  @spaces.GPU
620
  def save_pretrained(path):
621
  pro.save_pretrained(path)
622
+
623
+ pro.load_model()
624
  return 'save_pretrained'
625
 
626
  with gr.Blocks() as interface: