Upload app.py

app.py

@@ -342,8 +342,7 @@ def infer_brain_cf(*args):
         cf_[k] = (out["cf_pa"][k].mean(0).item() + 1) / 2 * (k_max - k_min) + k_min
     # plots
     # plt.close('all')
-    # effect = cf_x - rec_x
-    effect = cf_x - obs["x"]
+    effect = cf_x - rec_x
     effect = get_fig_arr(
         effect,
         cmap="RdBu_r",
@@ -370,6 +369,7 @@ def infer_chest_cf(*args):
     n_particles = 16
     # preprocessing
     obs = DATA[dataset_id]["test"].dataset.__getitem__(int(idx))
+    observation = obs['x']
     for k, v in obs.items():
         obs[k] = v.to(MODELS[dataset_id]["vae_args"].device).float()
         if n_particles > 1:
@@ -415,10 +415,8 @@ def infer_chest_cf(*args):
     cf_a = (cf_a.item() + 1) * 50
     # plots
     # plt.close('all')
-    effect = cf_x - rec_x
-    # effect = cf_x - obs["x"]
-    print(effect.max())
-    print(effect.min())
+    # effect = cf_x - rec_x
+    effect = cf_x - postprocess(observation)
     effect = get_fig_arr(
         effect,
         cmap="RdBu_r",