Update app.py

app.py

@@ -4,7 +4,7 @@ from datasets import load_dataset
 
 # Load the tokenizer and model
 tokenizer = T5Tokenizer.from_pretrained('t5-small')
-model = T5ForConditionalGeneration.from_pretrained('t5-small')
+model = T5ForConditionalGeneration.from_pretrained('t5-small', legacy=False)
 
 dataset = load_dataset("wikisql", split='test')
 for i in range(3):  # Let's take the first 3 examples