Update app.py
Browse files
app.py
CHANGED
@@ -18,12 +18,13 @@ disable_caching()
|
|
18 |
huggingface_token = os.environ["huggingface_token"]
|
19 |
pipe = pipeline(model="mskov/whisper-small-esc50")
|
20 |
print(pipe)
|
21 |
-
processor = WhisperProcessor.from_pretrained("
|
22 |
dataset = load_dataset("mskov/miso_test", split="test").cast_column("audio", Audio(sampling_rate=16000))
|
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 =
|
|
|
27 |
# Evaluate the model
|
28 |
# model.eval()
|
29 |
#print("model.eval ", model.eval())
|
|
|
18 |
huggingface_token = os.environ["huggingface_token"]
|
19 |
pipe = pipeline(model="mskov/whisper-small-esc50")
|
20 |
print(pipe)
|
21 |
+
processor = WhisperProcessor.from_pretrained("mskov/whisper-small-esc50")
|
22 |
dataset = load_dataset("mskov/miso_test", split="test").cast_column("audio", Audio(sampling_rate=16000))
|
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/miso_test").to("cuda")
|
27 |
+
|
28 |
# Evaluate the model
|
29 |
# model.eval()
|
30 |
#print("model.eval ", model.eval())
|