Upload app.py

app.py

@@ -406,10 +406,10 @@ def infer_chest_cf(*args):
     cf_a = out["cf_pa"]["age"].mean(0)
     # post process
     cf_x = postprocess(cf_x)
-    print(cf_x)
+    # print(cf_x[144,27])
     cf_x_std = cf_x_std.squeeze().detach().cpu().numpy()
     rec_x = postprocess(rec_x)
-    print(rec_x)
+    # print(rec_x[144,27])
     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,7 +417,8 @@ def infer_chest_cf(*args):
     # plots
     # plt.close('all')
     effect = cf_x - rec_x
-    print(effect)
+    print(effect.max())
+    print(effect.min())
     effect = get_fig_arr(
         effect,
         cmap="RdBu_r",