dev

app.py

@@ -243,10 +243,10 @@ def fine_tune_model(model, dataset_name, hub_id, api_key, num_epochs, batch_size
             # If the trainer_state.json is missing, look for the previous checkpoint
             print(f"Checkpoint {checkpoint_dir} is missing 'trainer_state.json'. Looking for previous checkpoints...")
             previous_checkpoints = sorted(glob.glob(os.path.join(os.path.dirname(checkpoint_dir), 'checkpoint-*')), key=os.path.getmtime)
-            
+            print(previous_checkpoints)
             if previous_checkpoints:
                 # Load the most recent previous checkpoint
-                last_checkpoint = previous_checkpoints[-1]
+                last_checkpoint = previous_checkpoints[-2]
                 print(f"Loading previous checkpoint: {last_checkpoint}")
                 train_result = trainer.train(resume_from_checkpoint=last_checkpoint)
             else: