mskov commited on
Commit
2512bd0
Β·
1 Parent(s): 7400537

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -31,7 +31,7 @@ model = WhisperForConditionalGeneration.from_pretrained("mskov/whisper-small-esc
31
  def map_to_pred(batch):
32
  audio = batch["audio"]
33
  input_features = processor(audio["array"], sampling_rate=audio["sampling_rate"], return_tensors="pt").input_features
34
- batch["reference"] = processor.tokenizer._normalize(batch['category'])
35
 
36
 
37
  with torch.no_grad():
 
31
  def map_to_pred(batch):
32
  audio = batch["audio"]
33
  input_features = processor(audio["array"], sampling_rate=audio["sampling_rate"], return_tensors="pt").input_features
34
+ batch["reference"] = processor.tokenizer._normalize(batch['audio'])
35
 
36
 
37
  with torch.no_grad():