Spaces:

hilamanor
/

audioEditing

Running on Zero

linoyts HF staff commited on Mar 4

Commit

e7fdc7d

•

1 Parent(s): a7153c1

limit input length

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -16,7 +16,7 @@ def load_audio(audio_path: Union[str, np.array], fn_STFT, left: int = 0, right:
         import audioldm
         import audioldm.audio
-        duration = audioldm.utils.get_duration(audio_path)
         mel, _, _ = audioldm.audio.wav_to_fbank(audio_path, target_length=int(duration * 102.4), fn_STFT=fn_STFT)
         mel = mel.unsqueeze(0)

         import audioldm
         import audioldm.audio
+        duration = min(audioldm.utils.get_duration(audio_path), 15)
         mel, _, _ = audioldm.audio.wav_to_fbank(audio_path, target_length=int(duration * 102.4), fn_STFT=fn_STFT)
         mel = mel.unsqueeze(0)