yes

app.py

@@ -67,7 +67,8 @@ def plot_heat(mat, title):
     plt.savefig(buf, format="png", bbox_inches='tight')
     buf.seek(0)
     plt.close(fig)
-    return buf
+    pil_image = Image.open(buf)
+    return pil_image
 
 # ─── SDXL Text Encoding ───────────────────────────────────────
 def encode_sdxl_prompt(prompt, negative_prompt=""):
@@ -341,7 +342,14 @@ with gr.Blocks(title="SDXL Dual Shunt Adapter", theme=gr.themes.Soft()) as demo:
             prompt, negative_prompt, adapter_l, adapter_g, strength, noise, gate_prob, 
             use_anchor, steps, cfg_scale, scheduler_name, width, height, seed
         ],
-        outputs=[out_img, delta_l, gate_l, delta_g, gate_g, stats_l, stats_g]
+        outputs=[
+            out_img, 
+            delta_l, 
+            gate_l, 
+            delta_g, 
+            gate_g, 
+            stats_l, 
+            stats_g]
     )
 
 if __name__ == "__main__":