Spaces:
Runtime error
Runtime error
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
|
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 =
|
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']}")
|