modify app

inference.py

@@ -155,13 +155,20 @@ class MasteringStyleTransfer:
         return data_tensor.to(self.device)
 
     def process_audio(self, input_audio, reference_audio, ito_reference_audio, params, perform_ito, log_ito=False):
+        print('run preprocess')
         input_tensor = self.preprocess_audio(input_audio, self.args.sample_rate)
         reference_tensor = self.preprocess_audio(reference_audio, self.args.sample_rate)
         ito_reference_tensor = self.preprocess_audio(ito_reference_audio, self.args.sample_rate)
+        print('preprocess done')
 
         reference_feature = self.get_reference_embedding(reference_tensor)
 
+        print('reference extracted')
+
         output_audio, predicted_params = self.mastering_style_transfer(input_tensor, reference_feature)
+
+        print('style transfer forward done')
+
         if perform_ito:
             ito_log = []
             for i in range(self.args.max_iter_ito):