progress bar fix

app.py

@@ -203,9 +203,18 @@ def generate_image(
         
         print(f"Starting generation with {num_inference_steps} steps...")
         
+        # Custom callback for progress updates
+        def progress_callback(pipe, step, timestep, callback_kwargs):
+            progress((0.2 + (step / num_inference_steps) * 0.75), 
+                    desc=f"Generating... Step {step}/{num_inference_steps}")
+            return callback_kwargs
+        
         # Generate image
         with torch.inference_mode():
-            result = pipeline(**pipe_kwargs)
+            result = pipeline(
+                **pipe_kwargs,
+                callback_on_step_end=progress_callback,
+            )
             image = result.images[0]
         
         print("Generation complete!")
@@ -260,7 +269,7 @@ with gr.Blocks(
                 label="Prompt",
                 placeholder="Describe the image you want to generate...",
                 lines=4,
-                value="Realistic macro photograph of a hermit crab using a soda can as its shell, partially emerging from the can, captured with sharp detail and natural colors, on a sunlit beach with soft shadows and a shallow depth of field, with blurred ocean waves in the background."
+                value="A cozy coffee shop scene on a rainy afternoon, warm lighting streaming through large windows with raindrops, a steaming cup of coffee on a wooden table with latte art, blurred background showing bookshelves and soft bokeh lights, photorealistic, cinematic composition, shallow depth of field"
             )
             
             image_input = gr.Image(