Update app.py

app.py

@@ -444,10 +444,11 @@ if uploaded_file:
                     # Phase 2
                     status_text.markdown("**Phase 2/4: Transcribing (Whisper)... This is the longest step.**")
                     compute_type = "float16" if device == "cuda" else "int8"
-                    # Default model size hardcoded to 'large-v2' to keep high quality while removing the sidebar toggle
+                    # Default model size hardcoded to 'large-v2'. 
+                    # CRITICAL FIX: batch_size lowered from 16 to 4 to prevent CUDA Out Of Memory on long files
                     model = whisperx.load_model("large-v2", device, compute_type=compute_type)
                     audio = whisperx.load_audio("temp_audio.wav")
-                    result = model.transcribe(audio, batch_size=16, language=target_language)
+                    result = model.transcribe(audio, batch_size=4, language=target_language)
                     del model
                     gc.collect()
                     torch.cuda.empty_cache()
@@ -524,4 +525,5 @@ if uploaded_file:
                         with st.expander("Show XML/EDL Code", expanded=False):
                             st.code(final_output, language="xml" if ext == "xml" else "text")
                         
-                        st.download_button(f"Download .{ext.upper()}", data=final_output, file_name=f"junior_editor_cut.{ext}")
+                        st.download_button(f"Download .{ext.upper()}", data=final_output, file_name=f"junior_editor_cut.{ext}")
+