lmoss commited on
Commit
91fb569
·
1 Parent(s): efaebf1

added dual pane view

Browse files
Files changed (1) hide show
  1. app.py +7 -2
app.py CHANGED
@@ -7,6 +7,8 @@ import time
7
  import numpy as np
8
  import streamlit.components.v1 as components
9
 
 
 
10
  url = "https://github.com/LukasMosser/PorousMediaGan/blob/master/checkpoints/berea/berea_generator_epoch_24.pth?raw=true"
11
 
12
  # If repo is private - we need to add a token in header:
@@ -19,7 +21,7 @@ with open('berea_generator_epoch_24.pth', 'wb') as f:
19
  st.text(resp.status_code)
20
 
21
  pv.set_plot_theme("document")
22
- pl = pv.Plotter(shape=(1, 1),
23
  window_size=(800, 800))
24
 
25
  netG = DCGAN3D_G(64, 512, 1, 32, 1)
@@ -49,7 +51,10 @@ grid.point_data['my_array'] = values
49
  slices = grid.slice_orthogonal()
50
  mesh = grid.contour(1, values, method='marching_cubes', rng=[1, 0], preference="points")
51
  dist = np.linalg.norm(mesh.points, axis=1)
52
- #pl.add_mesh(slices, cmap="gray")
 
 
 
53
  pl.add_mesh(mesh, scalars=dist)
54
  pl.export_html('pyvista.html')
55
 
 
7
  import numpy as np
8
  import streamlit.components.v1 as components
9
 
10
+ st.title("Generating Porous Media with GANs")
11
+
12
  url = "https://github.com/LukasMosser/PorousMediaGan/blob/master/checkpoints/berea/berea_generator_epoch_24.pth?raw=true"
13
 
14
  # If repo is private - we need to add a token in header:
 
21
  st.text(resp.status_code)
22
 
23
  pv.set_plot_theme("document")
24
+ pl = pv.Plotter(shape=(2, 1),
25
  window_size=(800, 800))
26
 
27
  netG = DCGAN3D_G(64, 512, 1, 32, 1)
 
51
  slices = grid.slice_orthogonal()
52
  mesh = grid.contour(1, values, method='marching_cubes', rng=[1, 0], preference="points")
53
  dist = np.linalg.norm(mesh.points, axis=1)
54
+
55
+ pl.add_mesh(slices, cmap="gray")
56
+ pl.subplot(1, 0)
57
+
58
  pl.add_mesh(mesh, scalars=dist)
59
  pl.export_html('pyvista.html')
60