Update app.py

app.py

@@ -1,25 +1,23 @@
 import gradio as gr
 import requests
 
-def search_huggingface_datasets(query, type):
-    if not query:  # If the query is empty, return an empty list
+def quick_search_query(query, type):
+    if not query:  
         return []
     url = f"https://huggingface.co/api/quicksearch?q={query}&type={type}&limit=20"
     response = requests.get(url)
     if response.status_code == 200:
         data = response.json()
         print(data)
-        dataset_names = [d['id'] for d in data['datasets']]
-        return dataset_names
+        repo_names = [d['id'] for d in data[type]]
+        return repo_names
     else:
-        return ["Error fetching datasets"]
+        return ["Error fetching repo"]
 
 def update_dropdown(query, type, key_up_data: gr.KeyUpData):
-    datasets = search_huggingface_datasets(key_up_data.input_value, type)
+    datasets = quick_search_query(key_up_data.input_value, type)
     return gr.update(choices=datasets, visible=True)
 
-
-
 with gr.Blocks() as demo:
     model_dropdown = gr.Dropdown(label="Models Auto-Complete", choices=[""], allow_custom_value=True)        
     dataset_dropdown = gr.Dropdown(label="Datasets Auto-Complete", choices=[""], allow_custom_value=True)