Spaces:
Runtime error
Runtime error
Commit
·
fa154e4
1
Parent(s):
29135e4
Update app.py
Browse files
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
|
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)
|