Update app.py

app.py

@@ -45,19 +45,20 @@ def generate(prompt, steps, guidance, seed, model_name):
     pipe = get_pipeline(model_id)
     generator = None if int(seed) == 0 else torch.manual_seed(int(seed))
 
-    # Ensure cross_attention_kwargs is always a dict (avoids NoneType errors)
+    # Ensure added_cond_kwargs is always a dict to avoid NoneType errors
     output = pipe(
         prompt=prompt,
         num_inference_steps=int(steps),
         guidance_scale=float(guidance),
         generator=generator,
         cross_attention_kwargs={},
+        added_cond_kwargs={},
     )
     return output.images
 
 # Build the Gradio UI
 with gr.Blocks() as demo:
-    gr.Markdown("## Model-Switcher Stable Diffusion Demo")
+    gr.Markdown("## ✨ Model-Switcher Stable Diffusion Demo")
     prompt     = gr.Textbox(label="Prompt", value="Retro robot in neon city")
     checkpoint = gr.Dropdown(choices=list(MODEL_OPTS.keys()),
                             value="SD v1.5 (base)",