Update app.py

app.py

@@ -1,5 +1,9 @@
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+from datasets import load_dataset
+
+# Load the Spider dataset
+spider_dataset = load_dataset("spider", split='train[:5]')
 
 # Load tokenizer and model
 tokenizer = AutoTokenizer.from_pretrained("mrm8488/t5-base-finetuned-wikiSQL")
@@ -23,4 +27,4 @@ interface = gr.Interface(
 
 # Launch the app
 if __name__ == "__main__":
-    interface.launch()
+    interface.launch()