frogcho123 commited on
Commit
fa154e4
·
1 Parent(s): 29135e4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ tokenizer = AutoTokenizer.from_pretrained("alirezamsh/small100")
12
  model = AutoModelForSeq2SeqLM.from_pretrained("alirezamsh/small100")
13
 
14
  def translate_speech(audio, target_lang):
15
- audio = audio[0].astype("float32") # Extract audio from tuple and convert to float32
16
  audio = whisper.pad_or_trim(audio, whisper_model.audio_config.sample_rate)
17
  mel = whisper.log_mel_spectrogram(audio).to(whisper_model.device)
18
  _, probs = whisper_model.detect_language(mel)
 
12
  model = AutoModelForSeq2SeqLM.from_pretrained("alirezamsh/small100")
13
 
14
  def translate_speech(audio, target_lang):
15
+ audio = audio.astype("float32") # Convert audio to float32
16
  audio = whisper.pad_or_trim(audio, whisper_model.audio_config.sample_rate)
17
  mel = whisper.log_mel_spectrogram(audio).to(whisper_model.device)
18
  _, probs = whisper_model.detect_language(mel)