small code cleanup

app.py

@@ -89,6 +89,9 @@ st.markdown(
     """
 , unsafe_allow_html=True)
 
+view_width = 400
+view_height = 400
+
 model_fname = "berea_generator_epoch_24.pth"
 checkpoint_url = "https://github.com/LukasMosser/PorousMediaGan/blob/master/checkpoints/berea/{0:}?raw=true".format(model_fname)
 
@@ -100,12 +103,12 @@ slices, mesh, dist = create_uniform_mesh_marching_cubes(img)
 
 pv.set_plot_theme("document")
 pl = pv.Plotter(shape=(1, 1),
-                     window_size=(400, 400))
+                     window_size=(view_width, view_height))
 _ = pl.add_mesh(slices, cmap="gray")
 pl.export_html('slices.html')
 
 pl = pv.Plotter(shape=(1, 1),
-                     window_size=(400, 400))
+                     window_size=(view_width, view_height))
 _ = pl.add_mesh(mesh, scalars=dist)
 pl.export_html('mesh.html')
 
@@ -113,8 +116,7 @@ st.header("2D Cross-Section of Generated Volume")
 fig = create_matplotlib_figure(img, img.shape[0]//2)
 st.pyplot(fig=fig)
 
-view_width = 400
-view_height = 400
+
 
 HtmlFile = open("slices.html", 'r', encoding='utf-8')
 source_code = HtmlFile.read()