mskov commited on
Commit
084283c
β€’
1 Parent(s): 62b1565

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -29,7 +29,7 @@ feature_extractor = AutoFeatureExtractor.from_pretrained("mskov/whisper_miso", u
29
 
30
  model_config = WhisperConfig.from_pretrained("mskov/whisper_miso", use_auth_token=huggingface_token)
31
  model = WhisperModel(config=model_config)
32
- model.load_state_dict(torch.load("whisper_miso/checkpoint-4000/rng_state.pth"))
33
  model.eval()
34
 
35
  dataset = load_dataset("mskov/miso_test", split="test").cast_column("audio", Audio())
 
29
 
30
  model_config = WhisperConfig.from_pretrained("mskov/whisper_miso", use_auth_token=huggingface_token)
31
  model = WhisperModel(config=model_config)
32
+ model.load_state_dict(torch.load("mskov/whisper_miso/pytorch_model.bin", use_auth_token=huggingface_token))
33
  model.eval()
34
 
35
  dataset = load_dataset("mskov/miso_test", split="test").cast_column("audio", Audio())