mskov commited on
Commit
40384d5
Β·
1 Parent(s): f547035

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -26,7 +26,7 @@ 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"]["array"], return_tensors="pt")
31
  input_features = inputs.input_features
32
  decoder_input_ids = torch.tensor([[1, 1]]) * model.config.decoder_start_token_id
 
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"]["array"], return_tensors="pt")
31
  input_features = inputs.input_features
32
  decoder_input_ids = torch.tensor([[1, 1]]) * model.config.decoder_start_token_id