Update app.py

app.py

@@ -40,7 +40,7 @@ def fine_tune_model(base_model_id, dataset):
 
     # Preprocess the dataset
     def preprocess(examples):
-        model_inputs = tokenizer(examples["text"], text_target=examples["sql"], max_length=512, truncation=True)
+        model_inputs = tokenizer(examples["sql_prompt"], text_target=examples["sql"], max_length=512, truncation=True)
         return model_inputs
     
     dataset = dataset.map(preprocess, batched=True)