Update app.py
Browse files
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
|
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
|