Update utils_viz.py

utils_viz.py

@@ -2,16 +2,24 @@
 import io
 import matplotlib.pyplot as plt
 
+# utils_viz.py
+import io
+import matplotlib.pyplot as plt
+from PIL import Image   # ← add this import
+
 def bar_topk(top_rows):
     labels = [r["name"] for r in top_rows]
     vals = [r["score"] for r in top_rows]
-    fig, ax = plt.subplots(figsize=(7,3))
+    fig, ax = plt.subplots(figsize=(7, 3))
     ax.barh(labels[::-1], vals[::-1])
     ax.set_xlabel("Score")
-    ax.set_xlim(0,100)
+    ax.set_xlim(0, 100)
     fig.tight_layout()
+
     bio = io.BytesIO()
     fig.savefig(bio, format="png", dpi=160, bbox_inches="tight")
     plt.close(fig)
     bio.seek(0)
-    return bio
+
+    # Return a PIL Image (what gr.Image(type="pil") wants)
+    return Image.open(bio).convert("RGB")