nateraw commited on
Commit
7ce4577
·
1 Parent(s): 04550e1

Update create_dataset.py

Browse files
Files changed (1) hide show
  1. create_dataset.py +4 -4
create_dataset.py CHANGED
@@ -90,14 +90,14 @@ def extract_vocal_demucs(model, filename, out_filename, sr=44100, device=None, s
90
 
91
 
92
  def main(
93
- clips_csv_filepath = "theovon.csv",
94
- character = "theovon",
95
  do_extract_vocals = False,
96
  whisper_size = "medium",
97
  # Where raw yt clips will be downloaded to
98
- dl_dir = "raw_data",
99
  # Where actual data will be organized
100
- data_dir = "prepared_data",
101
  ):
102
  dl_path = Path(dl_dir) / character
103
  dl_path.mkdir(exist_ok=True, parents=True)
 
90
 
91
 
92
  def main(
93
+ clips_csv_filepath = "data.csv",
94
+ character = "somebody",
95
  do_extract_vocals = False,
96
  whisper_size = "medium",
97
  # Where raw yt clips will be downloaded to
98
+ dl_dir = "downloads",
99
  # Where actual data will be organized
100
+ data_dir = "dataset_raw",
101
  ):
102
  dl_path = Path(dl_dir) / character
103
  dl_path.mkdir(exist_ok=True, parents=True)