multimodalart HF staff commited on
Commit
f0cefe4
·
verified ·
1 Parent(s): 0b14871

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -1,21 +1,21 @@
1
  import gradio as gr
2
  import requests
3
 
4
- def quick_search_query(query, type):
5
  if not query:
6
  return []
7
- url = f"https://huggingface.co/api/quicksearch?q={query}&type={type}&limit=20"
8
  response = requests.get(url)
9
  if response.status_code == 200:
10
  data = response.json()
11
  print(data)
12
- repo_names = [d['id'] for d in data[type]]
13
  return repo_names
14
  else:
15
  return ["Error fetching repo"]
16
 
17
- def update_dropdown(query, type, key_up_data: gr.KeyUpData):
18
- datasets = quick_search_query(key_up_data.input_value, type)
19
  return gr.update(choices=datasets, visible=True)
20
 
21
  with gr.Blocks() as demo:
 
1
  import gradio as gr
2
  import requests
3
 
4
+ def quick_search_query(query, repo_type):
5
  if not query:
6
  return []
7
+ url = f"https://huggingface.co/api/quicksearch?q={query}&type={repo_type}&limit=20"
8
  response = requests.get(url)
9
  if response.status_code == 200:
10
  data = response.json()
11
  print(data)
12
+ repo_names = [d['id'] for d in data[repo_type]]
13
  return repo_names
14
  else:
15
  return ["Error fetching repo"]
16
 
17
+ def update_dropdown(query, repo_type, key_up_data: gr.KeyUpData):
18
+ datasets = quick_search_query(key_up_data.input_value, repo_type)
19
  return gr.update(choices=datasets, visible=True)
20
 
21
  with gr.Blocks() as demo: