Update app.py

app.py

@@ -4,7 +4,7 @@ from datasets import load_dataset
 from difflib import get_close_matches
 
 # Load the WikiSQL dataset
-wikisql_dataset = load_dataset("wikisql", split='train[:100]')  # Increase the number of examples for better matching
+wikisql_dataset = load_dataset("wikisql", split='train')  # Increase the number of examples for better matching
 
 # Load tokenizer and model
 tokenizer = AutoTokenizer.from_pretrained("mrm8488/t5-base-finetuned-wikiSQL")