Update app.py

app.py

@@ -40,6 +40,12 @@ def generate(prompt, negative_prompt, num_inference_steps, guidance_scale, width
     neg_embeds, neg_pooled = compel(negative_prompt)
     
     progress(0.1, desc="Generating images")
+    
+    # Define proper callback for step end
+    def callback_on_step_end(pipe, i, t, callback_kwargs):
+        progress((i + 1) / num_inference_steps)
+        return callback_kwargs
+    
     images = pipe(
         prompt_embeds=embeds,
         pooled_prompt_embeds=pooled,
@@ -50,8 +56,7 @@ def generate(prompt, negative_prompt, num_inference_steps, guidance_scale, width
         width=width,
         height=height,
         num_images_per_prompt=num_samples,
-        callback_steps=1,  # Fixed: Added callback_steps parameter
-        callback_on_step_end=lambda i, t, latents: progress((i + 1) / num_inference_steps)
+        callback_on_step_end=callback_on_step_end
     ).images
     
     return images
@@ -106,7 +111,7 @@ css = """
 with gr.Blocks(css=css, theme=gr.themes.Soft()) as demo:
     gr.HTML("""
     <div class="header">
-        <h1>🎨 UnfilteredAI NSFW-gen-v2 Image Generator</h1>
+        <h1>🎨 Unfiltered AI NSFW-gen-v2 Image Generator</h1>
         <p>Enter creative prompts and generate high-quality images.</p>
     </div>
     """)