Update app.py

app.py

@@ -93,6 +93,7 @@ def fine_tune_model(model_name, dataset_name):
         args=training_args,
         train_dataset=train_dataset,
         eval_dataset=test_dataset,
+        tokenizer=tokenizer,
         compute_metrics=lambda pred: {"accuracy": torch.sum(pred.label_ids == pred.predictions.argmax(-1))},
     )
 
@@ -100,9 +101,10 @@ def fine_tune_model(model_name, dataset_name):
     print(trainer)
     print("###")
     
-    # Train model
+    # Train and save model
     #trainer.train()
-
+    #trainer.save_model()
+    
 def prompt_model(model_name, system_prompt, user_prompt, sql_schema):
     pipe = pipeline("text-generation", 
                     model=model_name,