Spaces:
Runtime error
Runtime error
import os | |
import click | |
import gin | |
from neural_waveshaping_synthesis.data.utils.create_dataset import create_dataset | |
from neural_waveshaping_synthesis.utils import seed_all | |
def get_filenames(directory): | |
return [os.path.join(directory, f) for f in os.listdir(directory) if ".wav" in f] | |
def main(gin_file, data_directory, output_directory, seed=0, device="cpu"): | |
gin.constant("device", device) | |
gin.parse_config_file(gin_file) | |
seed_all(seed) | |
files = get_filenames(data_directory) | |
create_dataset(files, output_directory) | |
if __name__ == "__main__": | |
main() |