Update app.py

app.py

@@ -36,7 +36,7 @@ def load_model_from_config(config, ckpt = None, verbose=True):
     else:
         print(f"Note chat no ckpt is loaded !!!")
 
-    # model.cuda()
+    model.cuda()
     model.eval()
     return model
 
@@ -83,7 +83,7 @@ class GenSamples:
 
             for idx,spec in enumerate(x_samples_ddim):
                 spec = spec.squeeze(0).cpu().numpy()
-                print(spec[0])
+                # print(spec[0])
                 record_dict = {'caption':prompt['ori_caption'][0]}
                 if self.save_mel:
                     mel_path = os.path.join(self.outpath,mel_name+f'_{idx}.npy')