Update app.py

app.py

@@ -19,7 +19,7 @@ huggingface_token = os.environ["huggingface_token"]
 pipe = pipeline(model="mskov/whisper-small-esc50") 
 print(pipe)
 processor = WhisperProcessor.from_pretrained("mskov/whisper-small-esc50")
-dataset = load_dataset("mskov/Misophonia_sounds", split="test").cast_column("audio", Audio(sampling_rate=16000))
+dataset = load_dataset("mskov/misophonia_sounds", split="test").cast_column("audio", Audio(sampling_rate=16000))
 
 print(dataset, "and at 0[audio][array] ", dataset[0]["audio"]["array"], type(dataset[0]["audio"]["array"]), "and at audio : ", dataset[0]["audio"])
 
@@ -31,7 +31,7 @@ model = WhisperForConditionalGeneration.from_pretrained("mskov/whisper-small-esc
 def map_to_pred(batch):
     audio = batch["audio"]
     input_features = processor(audio["array"], sampling_rate=audio["sampling_rate"], return_tensors="pt").input_features
-    batch["reference"] = processor.tokenizer._normalize(batch['category'])
+    batch["reference"] = processor.tokenizer._normalize(batch['sentence'])
     
     
     with torch.no_grad():