Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -4,14 +4,14 @@ import tempfile
|
|
| 4 |
|
| 5 |
# Function to transcribe audio and detect language
|
| 6 |
def transcribe_and_detect_language(audio_file):
|
| 7 |
-
model = whisper.load_model("base")
|
| 8 |
|
| 9 |
-
# Load
|
| 10 |
audio = whisper.load_audio(audio_file)
|
| 11 |
audio = whisper.pad_or_trim(audio)
|
| 12 |
|
| 13 |
-
#
|
| 14 |
-
mel = whisper.log_mel_spectrogram(audio).to(model.device)
|
| 15 |
|
| 16 |
# Detect the spoken language
|
| 17 |
_, probs = model.detect_language(mel)
|
|
|
|
| 4 |
|
| 5 |
# Function to transcribe audio and detect language
|
| 6 |
def transcribe_and_detect_language(audio_file):
|
| 7 |
+
model = whisper.load_model("base").to("cpu").float() # Ensure model is in full precision
|
| 8 |
|
| 9 |
+
# Load and process audio
|
| 10 |
audio = whisper.load_audio(audio_file)
|
| 11 |
audio = whisper.pad_or_trim(audio)
|
| 12 |
|
| 13 |
+
# Convert to log-Mel spectrogram in full precision
|
| 14 |
+
mel = whisper.log_mel_spectrogram(audio).to(model.device).float() # Convert to float32
|
| 15 |
|
| 16 |
# Detect the spoken language
|
| 17 |
_, probs = model.detect_language(mel)
|