multimodalart HF Staff commited on
Commit
4d93d41
·
verified ·
1 Parent(s): 8a8199e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -7
app.py CHANGED
@@ -2,26 +2,24 @@ import gradio as gr
2
  import requests
3
 
4
  def search_huggingface_datasets(query):
5
- print(query)
6
  if not query: # If the query is empty, return an empty list
7
  return []
8
  url = f"https://huggingface.co/api/quicksearch?q={query}&type=dataset&limit=20"
9
- print(url)
10
  response = requests.get(url)
11
  if response.status_code == 200:
12
  data = response.json()
13
- print(data)
14
  dataset_names = [d['id'] for d in data['datasets']]
15
  return dataset_names
16
  else:
17
  return ["Error fetching datasets"]
18
 
19
- def update_dropdown(query):
20
- datasets = search_huggingface_datasets(query)
21
- return gr.update(choices=datasets, visible=True)
 
22
 
23
  with gr.Blocks() as demo:
24
  with gr.Row():
25
  dataset_dropdown = gr.Dropdown(label="Datasets", choices=[""], elem_id="dataset_list", interactive=True, filterable=True, allow_custom_value=True)
26
- dataset_dropdown.key_up(update_dropdown, inputs=dataset_dropdown, outputs=dataset_dropdown, queue=False, show_progress="hidden")
27
  demo.launch()
 
2
  import requests
3
 
4
  def search_huggingface_datasets(query):
 
5
  if not query: # If the query is empty, return an empty list
6
  return []
7
  url = f"https://huggingface.co/api/quicksearch?q={query}&type=dataset&limit=20"
 
8
  response = requests.get(url)
9
  if response.status_code == 200:
10
  data = response.json()
 
11
  dataset_names = [d['id'] for d in data['datasets']]
12
  return dataset_names
13
  else:
14
  return ["Error fetching datasets"]
15
 
16
+ def update_dropdown(query, key_up_data: gr.KeyUpData):
17
+ datasets = search_huggingface_datasets(key_up_data.input_value)
18
+ print(key_up_data.input_value)
19
+ return gr.update(choices=datasets, value=key_up_data.input_value, visible=True)
20
 
21
  with gr.Blocks() as demo:
22
  with gr.Row():
23
  dataset_dropdown = gr.Dropdown(label="Datasets", choices=[""], elem_id="dataset_list", interactive=True, filterable=True, allow_custom_value=True)
24
+ dataset_dropdown.key_up(fn=update_dropdown, inputs=dataset_dropdown, outputs=dataset_dropdown, queue=False, show_progress="hidden")
25
  demo.launch()