Upload app.py

app.py

@@ -342,7 +342,8 @@ 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 - rec_x
+    effect = cf_x - obs["x"]
     effect = get_fig_arr(
         effect,
         cmap="RdBu_r",
@@ -406,10 +407,12 @@ def infer_chest_cf(*args):
     cf_a = out["cf_pa"]["age"].mean(0)
     # post process
     cf_x = postprocess(cf_x)
-    # print(cf_x[144,27])
+    print(cf_x.max())
+    print(cf_x.min())
     cf_x_std = cf_x_std.squeeze().detach().cpu().numpy()
     rec_x = postprocess(rec_x)
-    # print(rec_x[144,27])
+    print(rec_x.max())
+    print(rec_x.min())
     cf_r = RACE_CAT[cf_r.argmax(-1)]
     cf_s = SEX_CAT_CHEST[int(cf_s.item())]
     cf_f = FIND_CAT[cf_f.argmax(-1)] #FIND_CAT[int(cf_f.item())]
@@ -417,8 +420,8 @@ def infer_chest_cf(*args):
     # plots
     # plt.close('all')
     effect = cf_x - rec_x
-    print(effect.max())
-    print(effect.min())
+    # print(effect.max())
+    # print(effect.min())
     effect = get_fig_arr(
         effect,
         cmap="RdBu_r",