From ed16374ecee6f547c49fe4af0630294add317399 Mon Sep 17 00:00:00 2001
From: Lounger <4087076+TheLounger@users.noreply.github.com>
Date: Sun, 12 Jan 2025 03:35:22 +0100
Subject: [PATCH] Fix the gallery extension (#6656)

---
 extensions/gallery/script.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/extensions/gallery/script.py b/extensions/gallery/script.py
index ff0242c8..54f9c745 100644
--- a/extensions/gallery/script.py
+++ b/extensions/gallery/script.py
@@ -93,10 +93,11 @@ def generate_html():
 
 def filter_cards(filter_str=''):
     if filter_str == '':
-        return cards
+        return gr.Dataset(samples=cards)
 
     filter_upper = filter_str.upper()
-    return [k for k in cards if filter_upper in k[1].upper()]
+    filtered = [k for k in cards if filter_upper in k[1].upper()]
+    return gr.Dataset(samples=filtered)
 
 
 def select_character(evt: gr.SelectData):