mueller-franzes commited on
Commit
fec8e67
1 Parent(s): 0cb538a

add eye diffusion

Browse files
runs/2022_12_12_224914_eye_vaegan/last_vae.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d7ba4677d8c449e2aac91b3da7878c3e6b54225fb30313417b5507918809e532
3
+ size 112141527
runs/eye_diffusion/last.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cb831a7784cb5fdf6902e20832f1133a600973759f0618fb7e2a75129a569f98
3
+ size 2444594485
streamlit/pages/eye.py CHANGED
@@ -21,7 +21,7 @@ device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
21
 
22
  @st.cache(allow_output_mutation = True)
23
  def init_pipeline():
24
- pipeline = DiffusionPipeline.load_from_checkpoint('runs/2022_11_11_175610_eye_diffusion/last.ckpt')
25
  return pipeline
26
 
27
  if st.button('Sample'):
 
21
 
22
  @st.cache(allow_output_mutation = True)
23
  def init_pipeline():
24
+ pipeline = DiffusionPipeline.load_from_checkpoint('runs/eye_diffusion/last.ckpt')
25
  return pipeline
26
 
27
  if st.button('Sample'):