hafsaabd82 commited on
Commit
fbf7dc6
·
verified ·
1 Parent(s): abb0a4b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -198,15 +198,16 @@ def analyze_audio(audio_file: str,
198
  print(f"Loading Whisper model '{model_name}' on {device}...")
199
  model = whisperx.load_model(model_name, device, compute_type="float32")
200
  audio_loaded = whisperx.load_audio(audio_for_model)
 
 
 
201
  print("Transcribing audio...")
202
  result = model.transcribe(audio_loaded, batch_size=4, language="ur"
203
  )
204
- language_code = result.get("detected_language")
205
- results.languageCode = language_code
206
  full_text = " ".join([seg['text'] for seg in result.get("segments", [])]).strip()
207
  results.rawTranscriptionText = full_text
208
  aligned = {"segments": result["segments"]}
209
- print(f"Detected language: {language_code}. Aligning transcription...")
210
  align_model = None
211
  metadata = None
212
  if language_code not in global_align_model_cache:
 
198
  print(f"Loading Whisper model '{model_name}' on {device}...")
199
  model = whisperx.load_model(model_name, device, compute_type="float32")
200
  audio_loaded = whisperx.load_audio(audio_for_model)
201
+ lang_result = model.transcribe(audio_loaded, task="langdetect")
202
+ language_code_detected = lang_result["language"]
203
+ languageCode = language_code_detected
204
  print("Transcribing audio...")
205
  result = model.transcribe(audio_loaded, batch_size=4, language="ur"
206
  )
 
 
207
  full_text = " ".join([seg['text'] for seg in result.get("segments", [])]).strip()
208
  results.rawTranscriptionText = full_text
209
  aligned = {"segments": result["segments"]}
210
+ print(f"Detected language: {languageCode}. Aligning transcription...")
211
  align_model = None
212
  metadata = None
213
  if language_code not in global_align_model_cache: