radames commited on
Commit
52d5c1d
1 Parent(s): 2ce61d3

add-quicksearch (#1)

Browse files

- Create requirements.txt (3f4837160b8087b20b623ff6ef18426a154fd98b)
- replace textbox with quicksearch component (e6c11fa98daaef2b6a39e656f9acdeaca486f059)
- Update requirements.txt (a114861921b638d98d1b4c1b5b6aa78ee8146c0e)
- Update app.py (07d1806a8ff591cb659a8d9c7140b78e38995b8a)

Files changed (2) hide show
  1. app.py +12 -3
  2. requirements.txt +1 -0
app.py CHANGED
@@ -1,5 +1,6 @@
1
  import gradio as gr
2
  import yaml
 
3
 
4
  MARKDOWN_DESCRIPTION = """
5
  # mergekit config.yaml generator
@@ -53,12 +54,20 @@ with gr.Blocks() as demo:
53
  gr.Markdown(MARKDOWN_DESCRIPTION)
54
  with gr.Row():
55
  # model_name_input = gr.Textbox(label="Model Name", value="my-merge")
56
- model1_input = gr.Textbox(label="Model 1", value="BioMistral/BioMistral-7B")
 
 
 
 
 
57
  model1_layers_input = gr.Textbox(
58
  label="Model 1 Layer Range", placeholder="[start, end]", value="[0, 32]"
59
  )
60
- model2_input = gr.Textbox(
61
- label="Model 2", value="CorticalStack/pastiche-crown-clown-7b-dare-dpo"
 
 
 
62
  )
63
  model2_layers_input = gr.Textbox(
64
  label="Model 2 Layer Range", placeholder="[start, end]", value="[0, 32]"
 
1
  import gradio as gr
2
  import yaml
3
+ from gradio_huggingfacehub_search import HuggingfaceHubSearch
4
 
5
  MARKDOWN_DESCRIPTION = """
6
  # mergekit config.yaml generator
 
54
  gr.Markdown(MARKDOWN_DESCRIPTION)
55
  with gr.Row():
56
  # model_name_input = gr.Textbox(label="Model Name", value="my-merge")
57
+ model1_input = HuggingfaceHubSearch(
58
+ label="Model 1",
59
+ placeholder="Search for model 1 on Huggingface",
60
+ search_type="model",
61
+ value="BioMistral/BioMistral-7B"
62
+ )
63
  model1_layers_input = gr.Textbox(
64
  label="Model 1 Layer Range", placeholder="[start, end]", value="[0, 32]"
65
  )
66
+ model2_input = HuggingfaceHubSearch(
67
+ label="Model 2",
68
+ placeholder="Search for model 2 on Huggingface",
69
+ search_type="model",
70
+ value="CorticalStack/pastiche-crown-clown-7b-dare-dpo"
71
  )
72
  model2_layers_input = gr.Textbox(
73
  label="Model 2 Layer Range", placeholder="[start, end]", value="[0, 32]"
requirements.txt ADDED
@@ -0,0 +1 @@
 
 
1
+ gradio_huggingfacehub_search==0.0.3