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):