data

app.py

@@ -4,6 +4,7 @@ import torch
 import torch.nn as nn
 from einops import rearrange
 import matplotlib.pyplot as plt
+from copy import deepcopy
 
 pred_global = None
 
@@ -104,10 +105,11 @@ def text_fn(taxon, name):
 
 def thresh_fn(val):
     global pred_global
-    pred_global = pred_global > val
+    preds = deepcopy(pred_global)
+    preds = preds > val
     cmap = plt.get_cmap('Greens')
 
-    rgba_img = cmap(pred_global)
+    rgba_img = cmap(preds)
     rgb_img = np.delete(rgba_img, 3, 2)
     return rgb_img