Update app.py

app.py

@@ -452,6 +452,12 @@ class TrinerModelVITS:
         
     
         training_args= self.training_args
+        training_args.weight_kl=1
+        training_args.d_learning_rate=2e-4
+        training_args.learning_rate=2e-4
+        training_args.weight_mel=45
+        training_args.num_train_epochs=4
+        training_args.eval_steps=1000
 
         set_seed(training_args.seed)
         scaler = GradScaler(enabled=training_args.fp16)