Spaces:
Running
Running
Update app.py
Browse files
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}"
|