wip

app.py

@@ -3,6 +3,7 @@ import numpy as np
 import matplotlib.pyplot as plt
 from copy import deepcopy
 import pandas as pd
+from PIL import Image
 
 
 def get_index_of_element_containing_word(lst, word):
@@ -19,6 +20,8 @@ obs = df.drop_duplicates(subset=["species"])["species"].tolist()
 obs = list(sorted(obs))
 del df
 
+stl_base = Image.open("stl_base.png").convert("RGB")
+
 
 def update_fn(val):
     if val=="Class":
@@ -43,6 +46,8 @@ def text_fn(taxon, name):
 
     rgba_img = cmap(preds)
     rgb_img = np.delete(rgba_img, 3, 2)
+    blend = Image.blend(stl_base, Image.fromarray((rgb_img * 255).astype(np.uint8)), 0.5)
+    rgb_img = np.array(blend)
     #return gr.Image(preds, label="Predicted Heatmap", visible=True)
     return rgb_img