Update app.py

app.py

@@ -10,6 +10,11 @@ tokenizer = AutoTokenizer.from_pretrained("mrm8488/t5-base-finetuned-wikiSQL")
 model = AutoModelForSeq2SeqLM.from_pretrained("mrm8488/t5-base-finetuned-wikiSQL")
 
 def generate_sql_from_dataset(index):
+    # Ensure the index is within the range of the dataset
+    index = int(index)  # Convert to integer in case it's passed as a string
+    if index < 0 or index >= len(spider_dataset):
+        return "Invalid index. Please enter a number between 0 and {}.".format(len(spider_dataset) - 1), ""
+
     # Get the natural language query from the dataset
     query = spider_dataset[index]['question']
     input_text = "translate English to SQL: " + query