Upload app.py

app.py

@@ -406,15 +406,18 @@ def infer_chest_cf(*args):
     cf_a = out["cf_pa"]["age"].mean(0)
     # post process
     cf_x = postprocess(cf_x)
+    print(cf_x)
     cf_x_std = cf_x_std.squeeze().detach().cpu().numpy()
     rec_x = postprocess(rec_x)
+    print(rec_x)
     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) * 2
+    effect = cf_x - rec_x
+    print(effect)
     effect = get_fig_arr(
         effect,
         cmap="RdBu_r",