Update app.py

app.py

@@ -2,24 +2,18 @@ import gradio as gr
 from huggingface_hub import InferenceClient
 
 from datasets import load_dataset
-import duckdb
+import pandas
 
 global ds
 
 
 def load_new_dataset():
     gr.Info(message="Loading dataset...")
-    ds = load_dataset("fka/awesome-chatgpt-prompts", split="train")
+    df = pd.read_csv("hf://datasets/fka/awesome-chatgpt-prompts/prompts.csv")
 
 def run_query(input: str):
     try:
-        df_results = duckdb.sql(
-            query=f"""
-            SELECT *
-            FROM ds
-            WHERE act LIKE '%{input}%'
-            """
-        ).to_df()
+        df_results = df[df['act'] == input]
 
         logging_message = f"Results for '{input}' found."
     except Exception as e: