Upload app.py

app.py

@@ -407,21 +407,18 @@ def infer_chest_cf(*args):
     cf_a = out["cf_pa"]["age"].mean(0)
     # post process
     cf_x = postprocess(cf_x)
-    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.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())]
     cf_a = (cf_a.item() + 1) * 50
     # plots
     # plt.close('all')
-    effect = cf_x - rec_x
-    # print(effect.max())
-    # print(effect.min())
+    # effect = cf_x - rec_x
+    effect = cf_x - obs["x"]
+    print(effect.max())
+    print(effect.min())
     effect = get_fig_arr(
         effect,
         cmap="RdBu_r",