Update app.py

app.py

@@ -68,7 +68,7 @@ def on_change_event(app_state):
     if app_state and app_state['running'] and app_state['img']:
         img = app_state['img']
         step = app_state['step']
-        label = f'Reconstructed image from the latent state at step {step}. It will get better :)'
+        label = f'Reconstructed image from the latent state at step {step+1}. It will get better :)'
         print(f'Updating the image:! {app_state}')
         app_state['dots'] += 1
         app_state['dots'] = app_state['dots'] % 10
@@ -94,7 +94,7 @@ with gr.Blocks() as demo:
             app_state['img'] = res
             app_state['step'] = step
             app_state['img_list'].append(res)
-            app_state['status_msg'] = f'Generating step {step + 1} (zero based :))'
+            app_state['status_msg'] = f'Generating step {step + 2}'
         
         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)