(audio_file, cut_seconds=5)
| 95 | |
| 96 | |
| 97 | def trim_audio(audio_file, cut_seconds=5): |
| 98 | audio, sr = torchaudio.load(audio_file) |
| 99 | num_samples = cut_seconds * sr |
| 100 | cutted_audio = audio[:, :num_samples] |
| 101 | output_path = os.path.join(AUDIO_PROMPT_DIR, "audio_prompt_" + generate_filename() + ".wav") |
| 102 | torchaudio.save(output_path, cutted_audio, sr) |
| 103 | return output_path |
| 104 | |
| 105 | @spaces.GPU(duration=120) |
| 106 | def music_generation(args): |
no test coverage detected