Update app.py

app.py

@@ -570,12 +570,13 @@ def greet(text,id):
     parser = HfArgumentParser((ModelArguments, DataTrainingArguments, VITSTrainingArguments))
     json_file = os.path.abspath('VitsModelSplit/finetune_config_ara.json')
     model_args, data_args, training_args = parser.parse_json_file(json_file = json_file)
+    print('start')
     sgl=get_state_grad_loss(mel=True,
                            # generator=False,
                         #    discriminator=False,
                             duration=False)
     
-    
+    print(training_args)
     training_args.num_train_epochs=1000
     training_args.fp16=True
     training_args.eval_steps=300
@@ -585,9 +586,11 @@ def greet(text,id):
     training_args.weight_mel=45
     training_args.num_train_epochs=4
     training_args.eval_steps=1000
+    device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+    print(device)
     
     b=int(id)
-    device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+    
     ctrain_datasets,eval_dataset,full_generation_dataset=get_data_loader(train_dataset_dirs = train_dataset_dirs,
                                                               eval_dataset_dir = os.path.join(dataset_dir,'eval'),
                                                             full_generation_dir = os.path.join(dataset_dir,'full_generation'),