rahul7star commited on
Commit
4afda96
·
verified ·
1 Parent(s): e6cdf00

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -37,7 +37,8 @@ import json
37
  import yaml
38
  from slugify import slugify
39
 
40
-
 
41
 
42
 
43
 
@@ -522,6 +523,9 @@ def start_training(
522
  yaml.dump(config, f)
523
 
524
  # Simulate training
 
 
 
525
  print(f"[INFO] Starting training with config: {config_path}")
526
  print(json.dumps(config, indent=2))
527
  return f"Training started successfully with config: {config_path}"
 
37
  import yaml
38
  from slugify import slugify
39
 
40
+ sys.path.insert(0, "ai-toolkit")
41
+ from toolkit.job import get_job
42
 
43
 
44
 
 
523
  yaml.dump(config, f)
524
 
525
  # Simulate training
526
+ job = get_job(config_path)
527
+ job.run()
528
+ job.cleanup()
529
  print(f"[INFO] Starting training with config: {config_path}")
530
  print(json.dumps(config, indent=2))
531
  return f"Training started successfully with config: {config_path}"