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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -27,7 +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", tokenizer=miso_tokenizer)
31
 
32
  #miso_tokenizer = WhisperTokenizer.from_pretrained("openai/whisper-small", use_auth_token=huggingface_token)
33
 
@@ -42,7 +42,7 @@ results = task_evaluator.compute(
42
  model_or_pipeline=whisper_miso,
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
+ 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)
33
 
 
42
  model_or_pipeline=whisper_miso,
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,