Update app.py

app.py

@@ -35,7 +35,7 @@ import json
 import yaml
 from slugify import slugify
 
-from flux_train import start_training
+from flux_train_ui import start_training
 
 
 
@@ -533,18 +533,7 @@ def start_training(
 
 
 
-import os
-import uuid
-import json
-import shutil
-import yaml
-from pathlib import Path
-from fastapi import FastAPI, HTTPException
-from fastapi.responses import JSONResponse
-from huggingface_hub import hf_hub_download
-from flux_train import start_training
 
-app = FastAPI()
 
 @app.post("/train-from-hf")
 def auto_run_lora_from_repo():