Update app.py

app.py

@@ -89,8 +89,9 @@ with gr.Blocks() as demo:
             print (f'In Callback on {app_state} Done!')
         
         prompt = prompt + ' masterpiece charcoal pencil art lord of the rings illustration'
-        img = pipe(prompt, height=512, width=512, num_inference_steps=inference_steps, callback=callback, callback_steps=2)
+        img = pipe(prompt, height=512, width=512, num_inference_steps=inference_steps, callback=callback, callback_steps=3)
         app_state['running'] = False
+        app_state['img'] = None
         return gr.update(value=img.images[0], label='Generated image')
     
     app_state = gr.State({'img': None, 
@@ -108,12 +109,12 @@ with gr.Blocks() as demo:
     bt_make_image = gr.Button(f"Generate an image (takes about 10-15 minutes on CPU).", visible=False)
     
     img_description = gr.Markdown('Image generation takes some time'
-                                  ' but here you can see the what is generated from the latent state of the diffuser every few steps.'
-                                  ' Usually there is a significant improvement around step 15, that yields much better result')
+                                  ' but here you can see here what is generated from the latent state of the diffuser every few steps.'
+                                  ' Usually there is a significant improvement around step 12 that yields much better result')
     image = gr.Image(label='Illustration for your story', show_label=True)
     
     inference_steps = gr.Slider(5, 30, 
-                                value=15, 
+                                value=20, 
                                 step=1, 
                                 visible=True,
                                 label=f"Num inference steps (more steps makes a better image but takes more time)")