Update pipeline.py

pipeline.py

@@ -41,9 +41,7 @@ def inference(model, audio_file, text, output_file, device='cuda'):
             "condition": conditions,
         } 
 
-        sep_segment = model.ss_model(input_dict)["waveform"]
-
-        sep_segment = sep_segment.squeeze(0).squeeze(0).data.cpu().numpy()
+        sep_segment = model.ss_model.chunk_inference(input_dict)
 
         write(output_file, 32000, np.round(sep_segment * 32767).astype(np.int16))
         print(f'Write separated audio to [{output_file}]')