Update app.py
Browse files
app.py
CHANGED
@@ -30,7 +30,7 @@ def enhance_speech(audio_file):
|
|
30 |
|
31 |
# Prepare DNN input
|
32 |
Y = torch.unsqueeze(model._forward_transform(model._stft(y.to(args["device"]))), 0)
|
33 |
-
Y = pad_spec(Y, mode="
|
34 |
|
35 |
# Reverse sampling
|
36 |
sampler = model.get_pc_sampler(
|
|
|
30 |
|
31 |
# Prepare DNN input
|
32 |
Y = torch.unsqueeze(model._forward_transform(model._stft(y.to(args["device"]))), 0)
|
33 |
+
Y = pad_spec(Y, mode="zero_pad") # Use "zero_pad" mode for padding
|
34 |
|
35 |
# Reverse sampling
|
36 |
sampler = model.get_pc_sampler(
|