Kevin Fink commited on
Commit
74aed7b
·
1 Parent(s): cf05c78
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -235,8 +235,8 @@ def fine_tune_model(model, dataset_name, hub_id, api_key, num_epochs, batch_size
235
  # Fine-tune the model
236
  #if os.path.exists(training_args.output_dir) and os.listdir(training_args.output_dir):
237
  #train_result = trainer.train(resume_from_checkpoint=True)
238
- else:
239
- train_result = trainer.train()
240
  trainer.push_to_hub(commit_message="Training complete!")
241
  except Exception as e:
242
  return f"An error occurred: {str(e)}, TB: {traceback.format_exc()}"
 
235
  # Fine-tune the model
236
  #if os.path.exists(training_args.output_dir) and os.listdir(training_args.output_dir):
237
  #train_result = trainer.train(resume_from_checkpoint=True)
238
+ #else:
239
+ train_result = trainer.train()
240
  trainer.push_to_hub(commit_message="Training complete!")
241
  except Exception as e:
242
  return f"An error occurred: {str(e)}, TB: {traceback.format_exc()}"