Spaces:
Running
Running
Hendrik Schroeter
commited on
Force mono
Browse files
app.py
CHANGED
|
@@ -117,6 +117,9 @@ def demo_fn(
|
|
| 117 |
assert tmp is not None
|
| 118 |
sample, meta = tmp
|
| 119 |
sample = sample[..., : 10 * meta.sample_rate] # limit to 10 seconds
|
|
|
|
|
|
|
|
|
|
| 120 |
logger.info(f"Loaded sample with shape {sample.shape}")
|
| 121 |
if noise_fn is not None:
|
| 122 |
noise, _ = load_audio(noise_fn, sr) # type: ignore
|
|
|
|
| 117 |
assert tmp is not None
|
| 118 |
sample, meta = tmp
|
| 119 |
sample = sample[..., : 10 * meta.sample_rate] # limit to 10 seconds
|
| 120 |
+
if sample.dim() > 1 and sample.shape[0] > 1:
|
| 121 |
+
assert sample.shape[1] > sample.shape[2], f"Expecting channels first, but got {sample.shape}"
|
| 122 |
+
sample = sample.mean(dim=0, keepdim=True)
|
| 123 |
logger.info(f"Loaded sample with shape {sample.shape}")
|
| 124 |
if noise_fn is not None:
|
| 125 |
noise, _ = load_audio(noise_fn, sr) # type: ignore
|