mskov commited on
Commit
c4cba89
β€’
1 Parent(s): b525482

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -28,11 +28,11 @@ 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
- whisper_miso=WhisperModel.from_pretrained("openai/whisper-small", use_auth_token=huggingface_token, device_map="auto")
34
 
35
- #miso_tokenizer = WhisperTokenizer.from_pretrained("openai/whisper-small", use_auth_token=huggingface_token)
36
 
37
 
38
  task_evaluator = evaluator("automatic-speech-recognition")
 
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, tokenizer=miso_tokenizer, device_map="auto")
32
+
33
+ #whisper_miso=WhisperModel.from_pretrained("openai/whisper-small", use_auth_token=huggingface_token, device_map="auto")
34
 
 
35
 
 
36
 
37
 
38
  task_evaluator = evaluator("automatic-speech-recognition")