Update app.py

app.py

@@ -12,14 +12,7 @@ def transcribe(audio_file):
     # Transcribe with original Whisper
     model = whisper.load_model("large", device)
     result = model.transcribe(audio_file)
-
-    # Load alignment model and metadata
-    model_a, metadata = whisperx.load_align_model(language_code=result["language"], device=device)
-
-    # Align Whisper output
-    result_aligned = whisperx.align(result["segments"], model_a, metadata, audio_file, device)
-
-    return {"aligned": result_aligned["segments"], "word_segments": result_aligned["word_segments"]}
+    return result
 
 inputs = gr.Audio(sources="upload", type="filepath")
 outputs = gr.JSON()