Update app.py

app.py

@@ -72,7 +72,6 @@ def fine_tune_model(base_model_id, dataset):
         save_on_each_node=True,
         load_best_model_at_end=True,
         eval_strategy="steps",
-        gradient_checkpointing=True,
     )
     
     # Create Trainer instance