feat: use mono dialog

my_submission/hdemucs.py

@@ -64,6 +64,10 @@ class HDemucs(IdentitySeparationModel):
             self.hdemucs(mixed_sound_array).squeeze().permute(1, 2, 0).cpu().numpy()
         )
 
+        seperated[self.instruments_idx["dialog"]] = seperated[
+            self.instruments_idx["dialog"]
+        ].mean(axis=1, keepdims=True)
+
         # input_length = len(left_mixed_arr)
         separated_music_arrays = {}
         output_sample_rates = {}