firstkillday commited on
Commit
ed5a588
·
verified ·
1 Parent(s): e20ee88

Upload src/models/whisper/audio2feature.py with huggingface_hub

Browse files
src/models/whisper/audio2feature.py CHANGED
@@ -9,7 +9,7 @@ class Audio2Feature():
9
  def __init__(self,
10
  whisper_model_type="tiny",
11
  model_path="./models/whisper/tiny.pt",
12
- device="cpu")::
13
  self.whisper_model_type = whisper_model_type
14
  self.model = load_model(model_path, device=device) #
15
 
 
9
  def __init__(self,
10
  whisper_model_type="tiny",
11
  model_path="./models/whisper/tiny.pt",
12
+ device="cpu"):
13
  self.whisper_model_type = whisper_model_type
14
  self.model = load_model(model_path, device=device) #
15