j-tobias
commited on
Commit
·
e3bf44e
1
Parent(s):
f3d14a8
removed device
Browse files
model.py
CHANGED
@@ -17,7 +17,7 @@ class Model:
|
|
17 |
self.options = [
|
18 |
"openai/whisper-tiny.en",
|
19 |
"facebook/s2t-medium-librispeech-asr",
|
20 |
-
"nvidia/stt_en_fastconformer_ctc_large"
|
21 |
]
|
22 |
self.selected = None
|
23 |
self.pipeline = None
|
@@ -37,7 +37,7 @@ class Model:
|
|
37 |
raise ValueError(f"Selected Option is not a valid value, see: {self.options}")
|
38 |
|
39 |
if option == "openai/whisper-tiny.en":
|
40 |
-
self.pipeline = pipeline("automatic-speech-recognition", model="openai/whisper-tiny.en"
|
41 |
self.normalize = self.pipeline.tokenizer.normalize
|
42 |
|
43 |
elif option == "facebook/s2t-medium-librispeech-asr":
|
|
|
17 |
self.options = [
|
18 |
"openai/whisper-tiny.en",
|
19 |
"facebook/s2t-medium-librispeech-asr",
|
20 |
+
#"nvidia/stt_en_fastconformer_ctc_large"
|
21 |
]
|
22 |
self.selected = None
|
23 |
self.pipeline = None
|
|
|
37 |
raise ValueError(f"Selected Option is not a valid value, see: {self.options}")
|
38 |
|
39 |
if option == "openai/whisper-tiny.en":
|
40 |
+
self.pipeline = pipeline("automatic-speech-recognition", model="openai/whisper-tiny.en")
|
41 |
self.normalize = self.pipeline.tokenizer.normalize
|
42 |
|
43 |
elif option == "facebook/s2t-medium-librispeech-asr":
|