Update app.py

app.py

@@ -5,7 +5,7 @@ sys.path.append(os.path.dirname(os.path.abspath(__file__)))
 os.makedirs("/tmp/hf_cache", exist_ok=True)
 from huggingface_hub import whoami    
 os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1"
-
+from fastapi import HTTPException
 import spaces
 from fastapi import FastAPI, Query
 from huggingface_hub import list_repo_files, hf_hub_download, upload_file
@@ -16,6 +16,7 @@ from fastapi import FastAPI, UploadFile, File
 from fastapi.middleware.cors import CORSMiddleware
 from pathlib import Path
 from pathlib import Path
+from flux_train_ui import start_training
 import uuid
 import shutil
 import json
@@ -428,7 +429,7 @@ def recursive_update(d, u):
 
 
 @spaces.GPU(duration=50)
-def start_training(
+def start_training0(
     lora_name,
     concept_sentence,
     steps,