Spaces:

mskov
/

test

Runtime error

mskov commited on Jun 30, 2023

Commit

eec6bed

•

1 Parent(s): db2e209

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("mskov/whisper_miso/pytorch_model.bin"))
 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"))
 model.eval()
 dataset = load_dataset("mskov/miso_test", split="test").cast_column("audio", Audio())