mskov commited on
Commit
3925f62
Β·
1 Parent(s): 8849149

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -28,6 +28,7 @@ model = WhisperModel.from_pretrained("mskov/whisper_miso", use_auth_token=huggin
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
 
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
+ print(inputs)
32
  input_features = inputs.input_features
33
  decoder_input_ids = torch.tensor([[1, 1]]) * model.config.decoder_start_token_id
34
  last_hidden_state = model(input_features, decoder_input_ids=decoder_input_ids).last_hidden_state