Update app.py

app.py

@@ -68,6 +68,7 @@ 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