Update app.py
Browse files
app.py
CHANGED
@@ -23,7 +23,7 @@ dataset = load_dataset("mskov/miso_test", split="test").cast_column("audio", Aud
|
|
23 |
|
24 |
print(dataset, "and at 0[audio][array] ", dataset[0]["audio"]["array"], type(dataset[0]["audio"]["array"]), "and at audio : ", dataset[0]["audio"])
|
25 |
|
26 |
-
model = WhisperForConditionalGeneration.from_pretrained("mskov/
|
27 |
|
28 |
# Evaluate the model
|
29 |
# model.eval()
|
|
|
23 |
|
24 |
print(dataset, "and at 0[audio][array] ", dataset[0]["audio"]["array"], type(dataset[0]["audio"]["array"]), "and at audio : ", dataset[0]["audio"])
|
25 |
|
26 |
+
model = WhisperForConditionalGeneration.from_pretrained("mskov/whisper-small-esc50").to("cuda")
|
27 |
|
28 |
# Evaluate the model
|
29 |
# model.eval()
|