Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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: {
|
| 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:
|