Update app.py

app.py

@@ -32,10 +32,8 @@ def fine_tune_model(base_model_name, dataset_name):
 
     print("### Dataset")
     print(dataset)
-    print("### Train example")
+    print("### Example")
     print(dataset["train"][:1])
-    print("### Test example")
-    print(dataset["test"][:1])
     print("###")
     
     # Load model
@@ -57,10 +55,8 @@ def fine_tune_model(base_model_name, dataset_name):
 
     print("### Pre-processed dataset")
     print(dataset)
-    print("### Train example")
+    print("### Example")
     print(dataset["train"][:1])
-    print("### Test example")
-    print(dataset["test"][:1])
     print("###")
     
     # Split dataset into training and validation sets
@@ -138,8 +134,7 @@ def prompt_model(model_name, system_prompt, user_prompt, sql_schema):
 def load_model(model_name):
     model = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto")
     tokenizer = AutoTokenizer.from_pretrained(model_name)
-    tokenizer.padding_side = "right"
-    #tokenizer.pad_token = tokenizer.eos_token
+    tokenizer.pad_token = tokenizer.eos_token
     
     return model, tokenizer