Update app.py

app.py

@@ -3,7 +3,7 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 from datasets import load_dataset
 
 # Load the WikiSQL dataset
-wikisql_dataset = load_dataset("spider", split='train')  # Load a subset of the dataset
+wikisql_dataset = load_dataset("wikisql", split='train')  # Load a subset of the dataset
 
 # Extract schema information from the dataset
 table_names = set()