Details

app.py

@@ -20,12 +20,12 @@ def get_result(token, dataset, config, split, prompt) -> str:
 with gr.Blocks() as demo:
     gr.Markdown(" ## PandasAI demo using datasets library")
     gr.Markdown(" pandasai library https://github.com/gventuri/pandas-ai")
-    gr.Markdown(" datasets library https://huggingface.co/docs/datasets/index")
+    gr.Markdown(" datasets library https://huggingface.co/docs/datasets")
 
     hf_token = gr.Textbox(label="hugging face token", placeholder="hf_xxx", type="password")
-    dataset = gr.Textbox(label="dataset", placeholder="mstz/iris")
-    config = gr.Textbox(label="config", placeholder="iris")
-    split = gr.Textbox(label="split", placeholder="train")
+    dataset = gr.Textbox(label="dataset", placeholder="mstz/iris", value="mstz/iris")
+    config = gr.Textbox(label="config", placeholder="iris", value="iris")
+    split = gr.Textbox(label="split", placeholder="train", value="train")
     prompt = gr.Textbox(label="prompt (str)", placeholder="How many records do I have?. Give me the median of sepal_width. Show me the first 5 rows.")
     output = gr.Textbox(label="Output Box")
     get_result_button = gr.Button("Submit")