Update app.py

app.py

@@ -600,7 +600,9 @@ def init_training():
 
 @spaces.GPU
 def init_Starting():
-    globals pro
+    global pro
+    device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+
     pro=TrinerModelVITS(dir_model=dir_model,
                     path_training_args='VitsModelSplit/finetune_config_ara.json',
                     train_dataset_dirs = train_dataset_dirs,