Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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(
|
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 |
-
|
|
|
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:
|