Update app.py

app.py

@@ -5,7 +5,7 @@ from dotenv import load_dotenv
 from httpx import Client
 from huggingface_hub import HfApi
 import pandas as pd
-#from transformers import pipeline
+from gradio_huggingfacehub_search import HuggingfaceHubSearch
 import spaces
 from llama_cpp import Llama
 
@@ -22,7 +22,6 @@ headers = {
 }
 client = Client(headers=headers)
 api = HfApi(token=HF_TOKEN)
-#pipe = pipeline("text-generation", model="motherduckdb/DuckDB-NSQL-7B-v0.1", device="cuda")
 llama = Llama(
         model_path="DuckDB-NSQL-7B-v0.1-q8_0.gguf",
         n_ctx=2048,
@@ -102,7 +101,12 @@ def text2sql(dataset_name, query_input):
 
 with gr.Blocks() as demo:
     gr.Markdown("# 💫 Generate SQL queries based on a given text for your Hugging Face Dataset 💫")
-    dataset_name = gr.Textbox("jamescalam/world-cities-geo", label="Dataset Name")
+    dataset_name = HuggingfaceHubSearch(
+            label="Hub Dataset ID",
+            placeholder="Search for dataset id on Huggingface",
+            search_type="dataset",
+        )
+    # dataset_name = gr.Textbox("jamescalam/world-cities-geo", label="Dataset Name")
     query_input = gr.Textbox("Cities from Albania country", label="Ask something about your data")
     examples = [
                 ["Cities from Albania country"],