update gradio button

app.py

@@ -169,14 +169,18 @@ with gr.Blocks(title="Clinical Consultation Summarizer", theme=gr.themes.Soft())
     
     # Processing
     process_btn.click(
-        fn=lambda audio, token: load_models(token) or transcribe_and_diarize(audio, token),
+    fn=lambda audio, token: load_models(token),  # Just load models first
+    inputs=[audio_input, hf_token],
+    outputs=[transcript_output, gr.State()]
+    ).then(
+        fn=transcribe_and_diarize,
         inputs=[audio_input, hf_token],
         outputs=[transcript_output, gr.State()]
-    ).success(
+    ).then(
         fn=generate_soap_notes,
         inputs=[transcript_output, model_choice, anonymize_check],
         outputs=soap_output
-    ).success(
+    ).then(
         fn=extract_medical_entities,
         inputs=transcript_output,
         outputs=entity_output