Update app.py

app.py

@@ -40,7 +40,6 @@ 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 +424,11 @@ def recursive_update(d, u):
             d[k] = v
     return d
 
-def start_traininglocal(
+
+
+
+@spaces.GPU(duration=50)
+def start_training(
     lora_name,
     concept_sentence,
     steps,