Update app.py

app.py

@@ -4,6 +4,7 @@ import pandas as pd
 import duckdb
 import gradio as gr
 import matplotlib.pyplot as plt
+from PIL import Image
 import io
 
 duckdb_con = duckdb.connect(database=":memory:")
@@ -150,7 +151,7 @@ def run_benchmark(operation, df, repeats):
 
 
 # ----------------------------------------------------------
-# Chart generator (NEW)
+# Chart generator (PIL Image) — FIXED
 # ----------------------------------------------------------
 
 def generate_chart(result):
@@ -167,11 +168,13 @@ def generate_chart(result):
     plt.tight_layout()
     plt.savefig(buf, format="png")
     buf.seek(0)
-    return buf
+    plt.close(fig)
+
+    return Image.open(buf)
 
 
 # ----------------------------------------------------------
-# Formatting Results
+# Markdown Formatting
 # ----------------------------------------------------------
 
 def format_result(result):
@@ -209,7 +212,7 @@ theme = gr.themes.Soft(primary_hue="indigo", neutral_hue="slate")
 
 with gr.Blocks(title="DuckDB vs Pandas Benchmark", theme=theme) as demo:
 
-    gr.Markdown("# 🐼 vs 🦆 DuckDB vs Pandas Performance Playground")
+    gr.Markdown("# 🐼 vs 🦆 DuckDB vs Pandas — Performance Playground")
 
     with gr.Tabs():