mueller-franzes commited on
Commit
7053bac
1 Parent(s): 800153e

add cheXpert model

Browse files
.gitignore CHANGED
@@ -18,8 +18,8 @@
18
 
19
  /build/
20
 
21
-
22
  /results
23
  /scripts/local_trash
24
 
25
- !/media/**
 
18
 
19
  /build/
20
 
21
+ !/runs/**
22
  /results
23
  /scripts/local_trash
24
 
25
+ !/media/**
runs/2022_12_12_133315_chest_vaegan/last_vae.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fd8fb0b871c98271f36cc9a835d6b219705ffebca94d6f22eab4f44d199bb7dd
3
+ size 112371735
runs/chest_diffusion/last.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4e5e23bb9161b179251959fcbd009fca6fe017373253d7abb0ef4a0f2fdfd379
3
+ size 2444947637
streamlit/pages/chest.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_27_085654_chest_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/chest_diffusion/last.ckpt')
25
  return pipeline
26
 
27
  if st.button('Sample'):