didnt fix seems like runtime issues, also caching checkpoint now

app.py

@@ -104,7 +104,8 @@ def main():
     model_fname = "berea_generator_epoch_24.pth"
     checkpoint_url = "https://github.com/LukasMosser/PorousMediaGan/blob/master/checkpoints/berea/{0:}?raw=true".format(model_fname)
 
-    download_checkpoint(checkpoint_url, (DOWNLOADS_PATH / model_fname))
+    if not (DOWNLOADS_PATH / model_fname).exists():
+        download_checkpoint(checkpoint_url, (DOWNLOADS_PATH / model_fname))
 
     latent_size = st.slider("Latent Space Size z", min_value=1, max_value=5, step=1)
     img = generate_image((DOWNLOADS_PATH / model_fname), latent_size=latent_size)
@@ -125,8 +126,6 @@ def main():
     fig = create_matplotlib_figure(img, img.shape[0]//2)
     st.pyplot(fig=fig)
 
-
-
     HtmlFile = open((DOWNLOADS_PATH / 'slices.html'), 'r', encoding='utf-8')
     source_code = HtmlFile.read()
     st.header("3D Intersections")
@@ -151,8 +150,6 @@ def main():
         }```
         """)
 
-    #os.remove("slices.html")
-    #os.remove("mesh.html")
 
 if __name__ == "__main__":
-    main()
+    main()