mskov commited on
Commit
8849149
Β·
1 Parent(s): 32c0ce0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -26,8 +26,8 @@ huggingface_token = os.environ["huggingface_token"]
26
 
27
  model = WhisperModel.from_pretrained("mskov/whisper_miso", use_auth_token=huggingface_token)
28
  feature_extractor = AutoFeatureExtractor.from_pretrained("mskov/whisper_miso", use_auth_token=huggingface_token)
29
- ds = load_dataset("mskov/ESC50", split="test")
30
- inputs = feature_extractor(ds[0]["audio"]["bytes"], return_tensors="pt")
31
  input_features = inputs.input_features
32
  decoder_input_ids = torch.tensor([[1, 1]]) * model.config.decoder_start_token_id
33
  last_hidden_state = model(input_features, decoder_input_ids=decoder_input_ids).last_hidden_state
 
26
 
27
  model = WhisperModel.from_pretrained("mskov/whisper_miso", use_auth_token=huggingface_token)
28
  feature_extractor = AutoFeatureExtractor.from_pretrained("mskov/whisper_miso", use_auth_token=huggingface_token)
29
+ ds = load_dataset("mskov/miso_test", split="test")
30
+ inputs = feature_extractor(ds[0]["audio"]["array"], return_tensors="pt")
31
  input_features = inputs.input_features
32
  decoder_input_ids = torch.tensor([[1, 1]]) * model.config.decoder_start_token_id
33
  last_hidden_state = model(input_features, decoder_input_ids=decoder_input_ids).last_hidden_state