Updating

app.py

@@ -426,7 +426,7 @@ if st.checkbox("Generate Linear Interpolation"):
         figure_2[0:28, i * 28:(i + 1) * 28, ] = generated_image[:, :, -1]
         linear_predicted_interps.append(generated_image[:, :, -1])
 
-    st.image(figure_2, width=600)
+    st.image(figure_2, width=600, key='linear_interpolation')
 
     # Code to display a gif
     # interpolate_gif(decoder_model_boxes, latent_point_1, latent_point_2)
@@ -459,7 +459,7 @@ if st.checkbox("Generate Linear Interpolation"):
 
         if st.checkbox("Generate STL Model"):
             # Generate the STL File
-            time.sleep(5)  # Add a delay to control the update rate
+            time.sleep(15)  # Add a delay to control the update rate
             linear_stl = convert_sdf_to_stl(linear_sdf, threshold_divisor=threshold_divisor_input)
 
             # Download the STL File
@@ -542,7 +542,7 @@ if st.checkbox("Generate Mesh Interpolation"):
             figure_3[i * 28:(i + 1) * 28, j * 28:(j + 1) * 28, ] = generated_image[:, :, -1]
             mesh_predicted_interps.append(generated_image[:, :, -1])
 
-    st.image(figure_3, width=600)
+    st.image(figure_3, width=600, key='mesh_interpolation')
 
     # Code for generating the STL file
     st.subheader("Create an STL file from the extruded image!")
@@ -573,7 +573,7 @@ if st.checkbox("Generate Mesh Interpolation"):
             str((mesh_sdf_max - abs(mesh_sdf_min)) / 2))
 
         if st.checkbox("Generate STL Model"):
-            time.sleep(5)  # Add a delay to control the update rate
+            time.sleep(15)  # Add a delay to control the update rate
             # Generate the STL File
             linear_stl = convert_sdf_to_stl(mesh_sdf, threshold_divisor=mesh_threshold_divisor_input)