RamAnanth1 commited on
Commit
d297f8f
1 Parent(s): f9230c7

Update downloader/whisper_post_processor.py

Browse files
downloader/whisper_post_processor.py CHANGED
@@ -5,7 +5,7 @@ from datasets import Dataset
5
  import re
6
 
7
  class WhisperPP(PostProcessor):
8
- def __init__(self,data,**whisper_options):
9
  super().__init__()
10
  self._options = whisper_options
11
  interpreter = WhisperInterpreter(self._options.pop("model_size","base"))
@@ -13,7 +13,7 @@ class WhisperPP(PostProcessor):
13
  self._process = getattr(interpreter, self._options.pop("mode","transcribe"))
14
  self._write = self._options.pop("write")
15
  self.videos_to_process = self._options.pop("number_videos",0)
16
- self.repoId = self._get_name()
17
 
18
  def run(self, info):
19
  self.to_screen(f"Processing Video {info['id']}")
 
5
  import re
6
 
7
  class WhisperPP(PostProcessor):
8
+ def __init__(self,data,name, **whisper_options):
9
  super().__init__()
10
  self._options = whisper_options
11
  interpreter = WhisperInterpreter(self._options.pop("model_size","base"))
 
13
  self._process = getattr(interpreter, self._options.pop("mode","transcribe"))
14
  self._write = self._options.pop("write")
15
  self.videos_to_process = self._options.pop("number_videos",0)
16
+ self.repoId = name
17
 
18
  def run(self, info):
19
  self.to_screen(f"Processing Video {info['id']}")