Spaces:

mskov
/

test

Runtime error

mskov commited on Jun 30, 2023

Commit

084283c

•

1 Parent(s): 62b1565

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -29,7 +29,7 @@ feature_extractor = AutoFeatureExtractor.from_pretrained("mskov/whisper_miso", u
 model_config = WhisperConfig.from_pretrained("mskov/whisper_miso", use_auth_token=huggingface_token)
 model = WhisperModel(config=model_config)
-model.load_state_dict(torch.load("whisper_miso/checkpoint-4000/rng_state.pth"))
 model.eval()
 dataset = load_dataset("mskov/miso_test", split="test").cast_column("audio", Audio())

 model_config = WhisperConfig.from_pretrained("mskov/whisper_miso", use_auth_token=huggingface_token)
 model = WhisperModel(config=model_config)
+model.load_state_dict(torch.load("mskov/whisper_miso/pytorch_model.bin", use_auth_token=huggingface_token))
 model.eval()
 dataset = load_dataset("mskov/miso_test", split="test").cast_column("audio", Audio())