Update app.py

app.py

@@ -168,7 +168,7 @@ with gr.Blocks(css=css) as demo:
         b4 = gr.Button(value='Start')
         b4.click(start,
                  [b4, embs, ys, calibrate_prompts],
-                 [b1, b2, b3, b4, img, calibrate_prompts])
+                 [b1, b2, b3, b4, img, embs, ys, calibrate_prompts])
     with gr.Row():
         html = gr.HTML('''<div style='text-align:center; font-size:32'>You will callibrate for several prompts and then roam.</ div>''')