dev

app.py

@@ -243,12 +243,12 @@ def fine_tune_model(model, dataset_name, hub_id, api_key, num_epochs, batch_size
             
         checkpoint_dir = training_args.output_dir
         # If the trainer_state.json is missing, look for the previous checkpoint
-        previous_checkpoints = sorted(os.listdir("/data/results"), key=get_checkpoint_int, reverse=True)
+        previous_checkpoints = sorted(os.listdir("/data/results/checkpoints"), reverse=True)
         print(f'CHECKPOINTs: {previous_checkpoints}')
         for check in previous_checkpoints:
             try:
                 print(f"Removing previous checkpoint {check}")
-                shutil.rmtree(os.path.join('/data/results', check)) 
+                shutil.rmtree(os.path.join('/data/results/checkpoints', check)) 
             except:
                 pass        
         try: