Update app.py

app.py

@@ -6,7 +6,12 @@ from datasets import load_dataset
 tokenizer = T5Tokenizer.from_pretrained('t5-small')
 model = T5ForConditionalGeneration.from_pretrained('t5-small')
 
-dataset = load_dataset("wikisql", split='test')[:3]
+dataset = load_dataset("wikisql", split='test')
+for i in range(3):  # Let's take the first 3 examples
+    item = dataset[i]
+    question = item['question']
+    examples.append([question])
+
 def generate_sql(question):
     # Format the question for the model if needed. For example:
     # input_text = f"translate English to SQL: {question}"
@@ -22,7 +27,6 @@ def generate_sql(question):
     sql_query = tokenizer.decode(output_ids, skip_special_tokens=True)
     return sql_query
 
-examples = [[item['question']] for item in dataset]
     
 # Define the Gradio interface
 iface = gr.Interface(