add

app.py

@@ -524,6 +524,7 @@ class BaseTrainer(object):
 
     def _create_cuda_model(self):
         args = self.args
+        other_tools.load_checkpoints(self.model, args.test_ckpt, args.g_name)
         args.num_quantizers = 6
         args.shared_codebook =  False
         args.quantize_dropout_prob = 0.2
@@ -731,8 +732,6 @@ def syntalker(audio_path,sample_stratege):
 
     # return one intance of trainer
     trainer = BaseTrainer(args, ap = audio_path)
-    other_tools.load_checkpoints(trainer.model, args.test_ckpt, args.g_name)
-    
     result = trainer.test_demo(999)
     return result