data

app.py

@@ -106,12 +106,13 @@ def text_fn(taxon, name):
 def thresh_fn(val):
     global pred_global
     preds = deepcopy(pred_global)
-    preds = preds > val
+    preds[preds<val] = 0
+    preds[preds>=val] = 1
     cmap = plt.get_cmap('Greens')
 
     rgba_img = cmap(preds)
     rgb_img = np.delete(rgba_img, 3, 2)
-    return rgb_img, "Confidence Threshold: " + str(val)
+    return rgb_img, "Confidence Threshold: " + str(val) + str(np.mean(pred_global)) + str(np.mean(preds))
 
 with gr.Blocks() as demo:
     gr.Markdown(