Update app.py

app.py

@@ -386,6 +386,7 @@ def load_tools():
 
 
 class TrinerModelVITS:
+    KC=0
     def __init__(self,dir_model="",
                  path_training_args="",
                  train_dataset_dirs=[],
@@ -639,11 +640,12 @@ pro=TrinerModelVITS(dir_model=dir_model,
 
 @spaces.GPU(duration=30)
 def run_train_epoch(num):
+   TrinerModelVITS.KC+=1
    if num >0:
      pro.init_training()
      for i in range(num):
     # model.train(True)
-        return  pro.run_train_epoch()
+        return  pro.run_train_epoch() +f'- kc={TrinerModelVITS.KC}'
    else:
       pro.save_pretrained(pro.dir_model)
       pro.load_model()