Spaces:
Running
Running
Ffftdtd5dtft
commited on
Commit
•
1654627
1
Parent(s):
1c1ba7e
Update app.py
Browse files
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 |
-
|
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]
|