Update app.py

app.py

@@ -68,7 +68,9 @@ def generate_heatmap_image(model_entry):
     plt.savefig(buf, format="png")
     plt.close()
     buf.seek(0)
-    return buf.read()
+    # Convert the buffer into a PIL Image.
+    image = Image.open(buf).convert("RGB")
+    return image
 
 def generate_heatmaps(selected_model_names):
     """
@@ -78,8 +80,8 @@ def generate_heatmaps(selected_model_names):
     filtered_entries = [entry for entry in DATA if entry["Meta"]["Model Name"] in selected_model_names]
     images = []
     for entry in filtered_entries:
-        img_bytes = generate_heatmap_image(entry)
-        images.append(img_bytes)
+        img = generate_heatmap_image(entry)
+        images.append(img)
     return images
 
 # -------------------------------