Update app.py

app.py

@@ -40,6 +40,10 @@ def fine_tune_model(base_model_id, dataset):
     model = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto")    
     tokenizer = AutoTokenizer.from_pretrained(model_name)
 
+    if tokenizer.pad_token is None:
+        tokenizer.add_special_tokens({"pad_token": "[PAD]"})
+        model.resize_token_embeddings(len(tokenizer))
+    
     # Preprocess the dataset
     def preprocess(examples):
         model_inputs = tokenizer(examples["sql_prompt"], text_target=examples["sql"], max_length=512, padding="max_length", truncation=True)