mskov commited on
Commit
cb09111
β€’
1 Parent(s): 5aa44bd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -39,7 +39,8 @@ task_evaluator = evaluator("automatic-speech-recognition")
39
  #data = load_dataset("audiofolder", data_dir="mskov/miso_test")
40
  # 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"])
41
  #dataset = load_dataset("mskov/miso_test", split="test").cast_column("audio", Audio())
42
- dataset = load_dataset("mskov/miso_test", split="test")
 
43
  print(dataset)
44
  results = task_evaluator.compute(
45
  model_or_pipeline=whisper_miso,
@@ -47,7 +48,7 @@ results = task_evaluator.compute(
47
  data=dataset,
48
  tokenizer=miso_tokenizer,
49
  #feature_extractor=miso_feature_extractor,
50
- input_column=["path","file_name"],
51
  label_column="category",
52
  # device=None,
53
  strategy="simple",
 
39
  #data = load_dataset("audiofolder", data_dir="mskov/miso_test")
40
  # 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"])
41
  #dataset = load_dataset("mskov/miso_test", split="test").cast_column("audio", Audio())
42
+ # dataset = load_dataset("mskov/miso_test", split="test")
43
+ dataset = load_dataset("mskov/ESC50", split="test")
44
  print(dataset)
45
  results = task_evaluator.compute(
46
  model_or_pipeline=whisper_miso,
 
48
  data=dataset,
49
  tokenizer=miso_tokenizer,
50
  #feature_extractor=miso_feature_extractor,
51
+ input_column="audio",
52
  label_column="category",
53
  # device=None,
54
  strategy="simple",