Update src/webui.py

src/webui.py

@@ -174,16 +174,25 @@ if __name__ == '__main__':
         )
         gr.Markdown("Duplicate the old CPU space for use in private: [![Duplicate this Space](https://huggingface.co/datasets/huggingface/badges/raw/main/duplicate-this-space-sm-dark.svg)](https://huggingface.co/spaces/r3gm/AICoverGen_old_stable_cpu?duplicate=true)\n\n") 
 
-        with gr.Tab("Find Models"):
-            model_query = gr.Textbox(placeholder="Eg: Taylor Swift, Drake", max_lines=1, lines=1)
-            model_display = gr.Dataframe(
-                headers=["name", "author", "url"],
-                datatype=["str", "str", "str"],
-                row_count=5,
-                col_count=(3, "fixed"),
-            )
-            filter_btn = gr.Button("Search")
-            filter_btn.click(lambda query: filter_models([], query), inputs=model_query, outputs=model_display)
+    with gr.Tab("Find Models"):
+        # Input field for model query
+        model_query = gr.Textbox(placeholder="Eg: Taylor Swift, Drake", max_lines=1, lines=1, label="Search Models")
+
+        # Dataframe to display the filtered models
+        model_display = gr.Dataframe(
+            headers=["name", "author", "url"],
+            datatype=["str", "str", "str"],
+            row_count=5,
+            col_count=(3, "fixed"),
+            label="Model Results"
+        )
+
+        # Button to trigger the search
+        filter_btn = gr.Button("Search")
+
+        # Connect the button click to the filter_models function
+        filter_btn.click(fn=lambda query: filter_models([], query), inputs=model_query, outputs=model_display)
+
             
         with gr.Tab("Generate"):