Update autorun_lora_gradio.py

autorun_lora_gradio.py

@@ -3,7 +3,7 @@ import uuid
 import gradio as gr
 from pathlib import Path
 from huggingface_hub import hf_hub_download
-from your_existing_training_file import create_dataset, start_training  # <-- update this import as needed
+from flux_train_ui import create_dataset, start_training  # <-- update this import as needed
 
 # Constants
 REPO_ID = "rahul7star/ohamlab"