Update app.py
Browse files
app.py
CHANGED
|
@@ -21,13 +21,16 @@ disable_caching()
|
|
| 21 |
|
| 22 |
p = pipeline("automatic-speech-recognition")
|
| 23 |
|
|
|
|
|
|
|
|
|
|
| 24 |
task_evaluator = evaluator("automatic-speech-recognition")
|
| 25 |
#url = {"test" : "https://huggingface.co/datasets/mskov/miso_test/blob/main/test_set.parquet"}
|
| 26 |
#data = load_dataset("audiofolder", data_dir="mskov/miso_test")
|
| 27 |
# data = load_dataset("audiofolder", data_files=["datasets/mskov/miso_test/test_set/and.wav","mskov/miso_test/test_set/chew1.wav","mskov/miso_test/test_set/chew3.wav", "mskov/miso_test/test_set/chew3.wav","mskov/miso_test/test_set/chew4.wav","mskov/miso_test/test_set/cough1.wav","mskov/miso_test/test_set/cough2.wav","mskov/miso_test/test_set/cough3.wav","mskov/miso_test/test_set/hi.wav","mskov/miso_test/test_set/knock_knock.wav","mskov/miso_test/test_set/mouth_sounds1.wav","mskov/miso_test/test_set/mouth_sounds2.wav","mskov/miso_test/test_set/no.wav","mskov/miso_test/test_set/not_bad.wav","mskov/miso_test/test_set/oh_i_wish.wav","mskov/miso_test/test_set/pop1.wav","mskov/miso_test/test_set/really.wav","mskov/miso_test/test_set/sigh1.wav","mskov/miso_test/test_set/sigh2.wav","mskov/miso_test/test_set/slurp1.wav","mskov/miso_test/test_set/slurp2.wav","mskov/miso_test/test_set/sneeze1.wav","mskov/miso_test/test_set/sneeze2.wav","mskov/miso_test/test_set/so_i_did_it_again.wav"])
|
| 28 |
dataset = load_dataset("mskov/miso_test", split="test").cast_column("audio", Audio())
|
| 29 |
results = task_evaluator.compute(
|
| 30 |
-
model_or_pipeline=
|
| 31 |
#model_or_pipeline="mskov/whisper-small.en",
|
| 32 |
data=dataset,
|
| 33 |
input_column="audio",
|
|
|
|
| 21 |
|
| 22 |
p = pipeline("automatic-speech-recognition")
|
| 23 |
|
| 24 |
+
whisper_miso=WhisperForConditionalGeneration.from_pretrained("mskov/whisper_miso", use_auth_token=huggingface_token)
|
| 25 |
+
|
| 26 |
+
|
| 27 |
task_evaluator = evaluator("automatic-speech-recognition")
|
| 28 |
#url = {"test" : "https://huggingface.co/datasets/mskov/miso_test/blob/main/test_set.parquet"}
|
| 29 |
#data = load_dataset("audiofolder", data_dir="mskov/miso_test")
|
| 30 |
# data = load_dataset("audiofolder", data_files=["datasets/mskov/miso_test/test_set/and.wav","mskov/miso_test/test_set/chew1.wav","mskov/miso_test/test_set/chew3.wav", "mskov/miso_test/test_set/chew3.wav","mskov/miso_test/test_set/chew4.wav","mskov/miso_test/test_set/cough1.wav","mskov/miso_test/test_set/cough2.wav","mskov/miso_test/test_set/cough3.wav","mskov/miso_test/test_set/hi.wav","mskov/miso_test/test_set/knock_knock.wav","mskov/miso_test/test_set/mouth_sounds1.wav","mskov/miso_test/test_set/mouth_sounds2.wav","mskov/miso_test/test_set/no.wav","mskov/miso_test/test_set/not_bad.wav","mskov/miso_test/test_set/oh_i_wish.wav","mskov/miso_test/test_set/pop1.wav","mskov/miso_test/test_set/really.wav","mskov/miso_test/test_set/sigh1.wav","mskov/miso_test/test_set/sigh2.wav","mskov/miso_test/test_set/slurp1.wav","mskov/miso_test/test_set/slurp2.wav","mskov/miso_test/test_set/sneeze1.wav","mskov/miso_test/test_set/sneeze2.wav","mskov/miso_test/test_set/so_i_did_it_again.wav"])
|
| 31 |
dataset = load_dataset("mskov/miso_test", split="test").cast_column("audio", Audio())
|
| 32 |
results = task_evaluator.compute(
|
| 33 |
+
model_or_pipeline=whisper_miso,
|
| 34 |
#model_or_pipeline="mskov/whisper-small.en",
|
| 35 |
data=dataset,
|
| 36 |
input_column="audio",
|