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}"
|