Two cleanups

app.py

@@ -303,7 +303,7 @@ with gradio.Blocks() as demo:
             d1 = gradio.Dropdown(density_options, label="Density 1", value=random.choice(density_options))
             th1 = gradio.Dropdown(thickness_options, label="Thickness 1", value=random.choice(thickness_options))
         with gradio.Column():
-            img1 = gradio.Image(label="Endpoint 2")
+            img1 = gradio.Image(label="Endpoint 1")
         with gradio.Column():
             t2 = gradio.Dropdown(endpoint_types, label="Type 2", value=random.choice(endpoint_types))
             d2 = gradio.Dropdown(density_options, label="Density 2", value=random.choice(density_options))
@@ -314,15 +314,8 @@ with gradio.Blocks() as demo:
         lattice_inputs_1 = [t1, d1, th1]
         [x.change(fn=generate_unit_cell, inputs=lattice_inputs_1, outputs=[img1]) for x in lattice_inputs_1]
         lattice_inputs_2 = [t2, d2, th2]
-        [x.change(fn=generate_unit_cell, inputs=lattice_inputs_1, outputs=[img1]) for x in lattice_inputs_2]
-
-        # t1.change(fn=generate_unit_cell, inputs=[t1, d1, th1], outputs=[img1])
-        # d1.change(fn=generate_unit_cell, inputs=[t1, d1, th1], outputs=[img1])
-        # th1.change(fn=generate_unit_cell, inputs=[t1, d1, th1], 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])
+        [x.change(fn=generate_unit_cell, inputs=lattice_inputs_2, outputs=[img2]) for x in lattice_inputs_2]
+
     steps = gradio.Dropdown(interpolation_options, label="Interpolation Length", value=random.choice(interpolation_options))
     btn = gradio.Button("Run")
     img = gradio.Image(label="Transition")