Update app.py

app.py

@@ -40,7 +40,7 @@ from slugify import slugify
 sys.path.insert(0, "ai-toolkit")
 from toolkit.job import get_job
 
-
+from flux_train_ui import start_training
 
 app = FastAPI()
 
@@ -425,7 +425,7 @@ def recursive_update(d, u):
             d[k] = v
     return d
 
-def start_training(
+def start_traininglocal(
     lora_name,
     concept_sentence,
     steps,