Update app.py

app.py

@@ -422,7 +422,7 @@ class TrinerModelVITS:
         
         
         self.initialize_training_components()
-        self.epoch_count=0
+    #    self.epoch_count=0
         
   
     def load_model(self):
@@ -594,13 +594,14 @@ pro=TrinerModelVITS(dir_model=dir_model,
 
 @spaces.GPU(duration=120)
 def run_train_epoch(num):
-   pro.init_training()
-   for i in range(num):
+   if num >0:
+     pro.init_training()
+     for i in range(num):
     # model.train(True)
         yield pro.run_train_epoch()
-    
-   pro.save_pretrained(pro.dir_model)
-   pro.load_model()
+   else:
+      pro.save_pretrained(pro.dir_model)
+      pro.load_model()
    return 'save model '
        
 @spaces.GPU