add-quicksearch

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 import yaml
+from gradio_huggingfacehub_search import HuggingfaceHubSearch
 
 MARKDOWN_DESCRIPTION = """
 # mergekit config.yaml generator
@@ -53,12 +54,20 @@ with gr.Blocks() as demo:
     gr.Markdown(MARKDOWN_DESCRIPTION)
     with gr.Row():
         # model_name_input = gr.Textbox(label="Model Name", value="my-merge")
-        model1_input = gr.Textbox(label="Model 1", value="BioMistral/BioMistral-7B")
+        model1_input = HuggingfaceHubSearch(
+                label="Model 1",
+                placeholder="Search for model 1 on Huggingface",
+                search_type="model",
+                value="BioMistral/BioMistral-7B"
+        )
         model1_layers_input = gr.Textbox(
             label="Model 1 Layer Range", placeholder="[start, end]", value="[0, 32]"
         )
-        model2_input = gr.Textbox(
-            label="Model 2", value="CorticalStack/pastiche-crown-clown-7b-dare-dpo"
+        model2_input = HuggingfaceHubSearch(
+                label="Model 2",
+                placeholder="Search for model 2 on Huggingface",
+                search_type="model",
+                value="CorticalStack/pastiche-crown-clown-7b-dare-dpo"
         )
         model2_layers_input = gr.Textbox(
             label="Model 2 Layer Range", placeholder="[start, end]", value="[0, 32]"

requirements.txt

@@ -0,0 +1 @@
+gradio_huggingfacehub_search==0.0.3