Update app.py

app.py

@@ -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['path'])
     
     
     with torch.no_grad():