added berea path download

app.py

@@ -1,29 +1,32 @@
 import streamlit as st
-import streamlit.components.v1 as components
 import pyvista as pv
-from pyvista import examples
-import numpy as np
 from dcgan import DCGAN3D_G
 import torch
 import requests
+import time
 
-url = "https://raw.githubusercontent.com/LukasMosser/PorousMediaGan/raw/master/checkpoints/berea/berea_generator_epoch_24.pth"
+url = "https://github.com/LukasMosser/PorousMediaGan/blob/master/checkpoints/berea/berea_generator_epoch_24.pth?raw=true"
 
 # If repo is private - we need to add a token in header:
 resp = requests.get(url)
+
+with open('berea_generator_epoch_24.pth', 'wb') as f:
+    f.write(resp.content)
+    time.sleep(5)
+
 print(resp.status_code)
 st.text(resp.status_code)
+
 pv.set_plot_theme("document")
 pl = pv.Plotter(shape=(1, 1),
                      window_size=(800, 800))
-
+print(torch.load("berea_generator_epoch_24.pth"))
 netG = DCGAN3D_G(64, 512, 1, 32, 1)
 netG.load_state_dict(torch.load("berea_generator_epoch_24.pth"))
 z = torch.randn(1, 512, 5, 5, 5)
 with torch.no_grad():
     X = netG(z)
-print(X.size())
-print(X.min(), X.max())
+
 st.image((X[0, 0, 32].numpy()+1)/2, output_format="png")
 """
 data = examples.load_channels()