Ffftdtd5dtft commited on
Commit
1654627
1 Parent(s): 1c1ba7e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -277,8 +277,7 @@ def transcribe_audio_from_redis(audio_file):
277
  processor = WhisperProcessor.from_pretrained("openai/whisper-small")
278
  model = WhisperForConditionalGeneration.from_pretrained("openai/whisper-small")
279
  model.config.forced_decoder_ids = None
280
- sample = load_dataset("hf-internal-testing/librispeech_asr_dummy", "clean", split="validation")[0]["audio"]
281
- input_features = processor(sample["array"], sampling_rate=sample["sampling_rate"], return_tensors="pt").input_features
282
  predicted_ids = model.generate(input_features)
283
  transcription = processor.batch_decode(predicted_ids, skip_special_tokens=True)
284
  return transcription[0]
 
277
  processor = WhisperProcessor.from_pretrained("openai/whisper-small")
278
  model = WhisperForConditionalGeneration.from_pretrained("openai/whisper-small")
279
  model.config.forced_decoder_ids = None
280
+ input_features = processor(audio_file, sampling_rate=16000, return_tensors="pt").input_features
 
281
  predicted_ids = model.generate(input_features)
282
  transcription = processor.batch_decode(predicted_ids, skip_special_tokens=True)
283
  return transcription[0]