Update app.py

app.py

@@ -7,7 +7,12 @@ from datasets import load_dataset
 
 # Load the dataset from Hugging Face Spaces
 dataset = load_dataset("Blessin/dialogues-one-liners")
-DIALOGUES = dataset["train"]["dialogue"]
+
+# Get the column name for the dialogue data
+column_name = dataset.column_names[0]
+
+# Access the dialogue data
+DIALOGUES = dataset["train"][column_name]
 
 def generate_statement():
     # Generate a random statement from the dataset