diff --git a/src/ui/tabs/audiogen_tab.py b/src/ui/tabs/audiogen_tab.py
index 78e1acd..f84c105 100644
--- a/src/ui/tabs/audiogen_tab.py
+++ b/src/ui/tabs/audiogen_tab.py
@@ -161,7 +161,7 @@ def create_audiogen_tab(
             prompt, variant, duration, temperature, cfg_coef, top_k, top_p, seed
         ):
             if not prompt:
-                return (
+                yield (
                     gr.update(value="Please enter a prompt"),
                     gr.update(),
                     gr.update(),
@@ -169,6 +169,7 @@ def create_audiogen_tab(
                     gr.update(),
                     gr.update(),
                 )
+                return
 
             yield (
                 gr.update(value="🔄 Generating..."),
diff --git a/src/ui/tabs/jasco_tab.py b/src/ui/tabs/jasco_tab.py
index bfa74b8..c382e0f 100644
--- a/src/ui/tabs/jasco_tab.py
+++ b/src/ui/tabs/jasco_tab.py
@@ -217,7 +217,7 @@ def create_jasco_tab(
             prompt, variant, chords, drums, duration, bpm, temperature, cfg_coef, top_k, top_p, seed
         ):
             if not chords:
-                return (
+                yield (
                     gr.update(value="Please enter a chord progression"),
                     gr.update(),
                     gr.update(),
@@ -225,6 +225,7 @@ def create_jasco_tab(
                     gr.update(),
                     gr.update(),
                 )
+                return
 
             yield (
                 gr.update(value="🔄 Generating..."),
diff --git a/src/ui/tabs/magnet_tab.py b/src/ui/tabs/magnet_tab.py
index f4312b9..c7516bb 100644
--- a/src/ui/tabs/magnet_tab.py
+++ b/src/ui/tabs/magnet_tab.py
@@ -184,7 +184,7 @@ def create_magnet_tab(
             prompt, variant, duration, temperature, cfg_coef, top_k, top_p, decoding_steps, span_arr, seed
         ):
             if not prompt:
-                return (
+                yield (
                     gr.update(value="Please enter a prompt"),
                     gr.update(),
                     gr.update(),
@@ -192,6 +192,7 @@ def create_magnet_tab(
                     gr.update(),
                     gr.update(),
                 )
+                return
 
             yield (
                 gr.update(value="🔄 Generating..."),
diff --git a/src/ui/tabs/musicgen_tab.py b/src/ui/tabs/musicgen_tab.py
index e6a26ce..c6dd751 100644
--- a/src/ui/tabs/musicgen_tab.py
+++ b/src/ui/tabs/musicgen_tab.py
@@ -189,7 +189,7 @@ def create_musicgen_tab(
             prompt, variant, duration, temperature, cfg_coef, top_k, top_p, seed, melody
         ):
             if not prompt:
-                return (
+                yield (
                     gr.update(value="Please enter a prompt"),
                     gr.update(),
                     gr.update(),
@@ -197,6 +197,7 @@ def create_musicgen_tab(
                     gr.update(),
                     gr.update(),
                 )
+                return
 
             # Update status
             yield (
diff --git a/src/ui/tabs/style_tab.py b/src/ui/tabs/style_tab.py
index 621d6d2..5bcdec9 100644
--- a/src/ui/tabs/style_tab.py
+++ b/src/ui/tabs/style_tab.py
@@ -161,7 +161,7 @@ def create_style_tab(
             prompt, variant, style_audio, duration, temperature, cfg_coef, top_k, top_p, seed
         ):
             if not style_audio:
-                return (
+                yield (
                     gr.update(value="Please upload a style reference audio"),
                     gr.update(),
                     gr.update(),
@@ -169,6 +169,7 @@ def create_style_tab(
                     gr.update(),
                     gr.update(),
                 )
+                return
 
             yield (
                 gr.update(value="🔄 Generating..."),