Update app.py

app.py

@@ -317,9 +317,9 @@ with gradio.Blocks() as demo:
             d2 = gradio.Dropdown(density_options, label="Density 2", value=random.choice(density_options))
             th2 = gradio.Dropdown(thickness_options, label="Thickness 2", value=random.choice(thickness_options))
             img2 = gradio.Image(label="Endpoint 2")
-            t2.change(fn=generate_unit_cell, inputs=[t2, d2, th2], outputs=[img1])
-            d2.change(fn=generate_unit_cell, inputs=[t2, d2, th2], outputs=[img1])
-            th2.change(fn=generate_unit_cell, inputs=[t2, d2, th2], outputs=[img1])
+            t2.change(fn=generate_unit_cell, inputs=[t2, d2, th2], outputs=[img2])
+            d2.change(fn=generate_unit_cell, inputs=[t2, d2, th2], outputs=[img2])
+            th2.change(fn=generate_unit_cell, inputs=[t2, d2, th2], outputs=[img2])
     steps = gradio.Dropdown(interpolation_options, label="Interpolation Length", value=random.choice(interpolation_options))
     btn = gradio.Button("Run")
     img = gradio.Image(label="Transition")