mskov commited on
Commit
703132a
β€’
1 Parent(s): 291a467

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -27,6 +27,7 @@ disable_caching()
27
  huggingface_token = os.environ["huggingface_token"]
28
 
29
  miso_tokenizer = WhisperTokenizer.from_pretrained("mskov/whisper_miso", use_auth_token=huggingface_token)
 
30
  whisper_miso=WhisperModel.from_pretrained("mskov/whisper_miso", use_auth_token=huggingface_token, device_map="auto")
31
 
32
  #miso_tokenizer = WhisperTokenizer.from_pretrained("openai/whisper-small", use_auth_token=huggingface_token)
@@ -43,6 +44,7 @@ results = task_evaluator.compute(
43
  #model_or_pipeline="mskov/whisper-small.en",
44
  data=dataset,
45
  tokenizer=miso_tokenizer,
 
46
  input_column="audio",
47
  label_column="audio",
48
  # device=None,
 
27
  huggingface_token = os.environ["huggingface_token"]
28
 
29
  miso_tokenizer = WhisperTokenizer.from_pretrained("mskov/whisper_miso", use_auth_token=huggingface_token)
30
+ miso_feature_extractor = WhisperFeatureExtractor.from_pretrained("mskov/whisper_miso", use_auth_token=huggingface_token)
31
  whisper_miso=WhisperModel.from_pretrained("mskov/whisper_miso", use_auth_token=huggingface_token, device_map="auto")
32
 
33
  #miso_tokenizer = WhisperTokenizer.from_pretrained("openai/whisper-small", use_auth_token=huggingface_token)
 
44
  #model_or_pipeline="mskov/whisper-small.en",
45
  data=dataset,
46
  tokenizer=miso_tokenizer,
47
+ feature_extractor=miso_feature_extractor,
48
  input_column="audio",
49
  label_column="audio",
50
  # device=None,