j-tobias commited on
Commit
e3bf44e
·
1 Parent(s): f3d14a8

removed device

Browse files
Files changed (1) hide show
  1. model.py +2 -2
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", device=0)
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":