Update app.py

app.py

@@ -91,11 +91,11 @@ def fine_tune_model(base_model_name, dataset_name):
     print(training_args)
     print("###")
 
-    # PEFT https://www.philschmid.de/fine-tune-flan-t5-peft
+    # PEFT
 
     lora_config = LoraConfig(
         r=16,
-        # TODO
+        # TODO https://www.philschmid.de/fine-tune-flan-t5-peft
         #bias="none",
         #lora_alpha=32,
         #lora_dropout=0.05,
@@ -118,11 +118,11 @@ def fine_tune_model(base_model_name, dataset_name):
 
     # Train model
     
-    #trainer.train()
+    trainer.train()
 
     # Push tokenizer to HF
 
-    #tokenizer.push_to_hub(FT_MODEL_NAME)
+    tokenizer.push_to_hub(FT_MODEL_NAME)
     
 def prompt_model(model_name, system_prompt, user_prompt, sql_context):
     pipe = pipeline("text-generation",