Update app.py

app.py

@@ -69,9 +69,9 @@ def on_change_event(app_state):
         step = app_state['step']
         label = f'Reconstructed image from the latent state at step {step}. It will get better :)'
         print(f'Updating the image:! {app_state}')
-        return gr.update(value=img, label=label), gr.update(value=app_state['img_list'], label='intermediate steps')
+        return gr.update(value=app_state['img_list'], label='intermediate steps')
     else:
-        return gr.update(label='Illustration will appear here soon'), gr.update(label='images list')
+        return gr.update(label='images list')
 
 with gr.Blocks() as demo:
     
@@ -91,7 +91,7 @@ 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=1)
+        img = pipe(prompt, height=512, width=512, num_inference_steps=inference_steps, callback=callback, callback_steps=2)
         app_state['running'] = False
         app_state['img'] = None
         return gr.update(value=img.images[0], label='Generated image')
@@ -113,10 +113,11 @@ 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 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)
+                                  ' but here you can see what is generated from the latent state of the diffuser every few steps.'
+                                  ' Usually there is a significant improvement around step 12 that yields a much better image')
     gallery = gr.Gallery()
+    image = gr.Image(label='Illustration for your story', show_label=True)
+
     gallery.style(grid=[4])
     
     inference_steps = gr.Slider(5, 30, 
@@ -130,12 +131,11 @@ with gr.Blocks() as demo:
     bt_make_image.click(fn=generate_image, inputs=[summary, inference_steps, app_state], outputs=image)
     
     eventslider = gr.Slider(visible=False)
-    dep = demo.load(on_change_event, app_state, [image, gallery], every=10)
-    eventslider.change(fn=on_change_event, inputs=[app_state], outputs=[image, gallery], every=10, cancels=[dep])
+    dep = demo.load(on_change_event, app_state, [gallery], every=10)
+    eventslider.change(fn=on_change_event, inputs=[app_state], outputs=[gallery], every=10, cancels=[dep])
     
 
 if READ_TOKEN:
     demo.queue().launch()
 else:
     demo.queue().launch(share=True, debug=True)
-