Spaces:

ylacombe
/

create-your-own-TTS-dataset

Paused

ylacombe commited on Dec 20, 2023

Commit

94eb218

1 Parent(s): 8a2eb00

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -88,7 +88,7 @@ def transcribe(inputs_path, task, use_demucs, dataset_name, oauth_token: gr.OAut
             audios.append(path)
             transcripts.append(chunk["text"])
-        dataset = Dataset.from_dict({"audio": audios, "transcript": transcripts}).cast_column("audio", Audio())
         current_step += 1
         progress((current_step, total_step), desc="Push dataset.")
@@ -195,7 +195,7 @@ def yt_transcribe(yt_url, task, use_demucs, dataset_name, oauth_token: gr.OAuthT
             audios.append(path)
             transcripts.append(chunk["text"])
-        dataset = Dataset.from_dict({"audio": audios, "transcript": transcripts}).cast_column("audio", Audio())
         current_step += 1
         progress((current_step, total_step), desc="Push dataset.")

             audios.append(path)
             transcripts.append(chunk["text"])
+        dataset = Dataset.from_dict({"audio": audios, "text": transcripts}).cast_column("audio", Audio())
         current_step += 1
         progress((current_step, total_step), desc="Push dataset.")
             audios.append(path)
             transcripts.append(chunk["text"])
+        dataset = Dataset.from_dict({"audio": audios, "text": transcripts}).cast_column("audio", Audio())
         current_step += 1
         progress((current_step, total_step), desc="Push dataset.")