Update dataset/transcript_dataset.py

dataset/transcript_dataset.py

@@ -10,6 +10,7 @@ from dataset.hf_dataset import HFDataset
 class TranscriptDataset(HFDataset):
 
   def __init__(self, name) -> None:
+    self.name = name
     super().__init__(name)
 
   def generate_dataset(self, input, download_path, overwrite, whisper_config):
@@ -25,7 +26,7 @@ class TranscriptDataset(HFDataset):
       #emptyDataset=self.dataset["train"].filter(lambda e: e["id"] is None)
       emptyDataset=self.dataset["train"]
     whisper_config["number_videos"] = 5
-    whisperPP = WhisperPP(emptyDataset, **whisper_config)
+    whisperPP = WhisperPP(emptyDataset, self.name, **whisper_config)
     downloader = YoutubeDownloader(download_path)
     if not overwrite:
       downloader.config["download_archive"] = os.path.join(download_path,"video_record.txt")