Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 from huggingface_hub import InferenceClient
 
 from datasets import load_dataset
+import duckdb
 
 global ds
 
@@ -77,7 +78,7 @@ def run_query(input: str):
         df_results = duckdb.sql(
             query=f"""
             SELECT *
-            FROM ds
+            FROM train
             WHERE act LIKE '%{input}%'
             """
         ).to_df()