Update app.py

app.py

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