Update app.py

app.py

@@ -70,7 +70,7 @@ if(is_spaces):
 def create_dataset(*inputs):
     print("Creating dataset")
     images = inputs[0]
-    destination_folder = str(uuid.uuid4())
+    destination_folder = str(f"datasets/{uuid.uuid4()}")
     if not os.path.exists(destination_folder):
         os.makedirs(destination_folder)
 
@@ -140,7 +140,7 @@ def start_training(
     slugged_lora_name = slugify(lora_name)
 
     # Load the default config
-    with open("config/examples/train_lora_flux_24gb.yaml", "r") as f:
+    with open("train_lora_flux_24gb.yaml", "r") as f:
         config = yaml.safe_load(f)
 
     # Update the config with user inputs
@@ -170,29 +170,17 @@ def start_training(
     config_path = f"config/{slugged_lora_name}.yaml"
     with open(config_path, "w") as f:
         yaml.dump(config, f)
-
-    job = get_job(config_path)
-
-    # Run the job
-    job.run()
-    job.cleanup()
+    if(is_spaces):
+        pass
+        #do the spacerunner things here
+    else:
+        #run the job locally
+        job = get_job(config_path)
+        job.run()
+        job.cleanup()
 
     return f"Training completed successfully. Model saved as {slugged_lora_name}"
 
-def start_training_spaces(
-    lora_name,
-    concept_sentence,
-    steps,
-    lr,
-    rank,
-    dataset_folder,
-    sample_1,
-    sample_2,
-    sample_3,
-):
-    #Feel free to include the spacerunner stuff here @abhishek
-    pass
-
 theme = gr.themes.Monochrome(
     text_size=gr.themes.Size(lg="18px", md="15px", sm="13px", xl="22px", xs="12px", xxl="24px", xxs="9px"),
     font=[gr.themes.GoogleFont('Source Sans Pro'), 'ui-sans-serif', 'system-ui', 'sans-serif'],
@@ -296,8 +284,9 @@ with gr.Blocks(theme=theme, css=css) as demo:
         ```bash
         git clone https://huggingface.co/spaces/flux-train/flux-lora-trainer
         cd flux-lora-trainer
+        pip install requirements_local.txt
         ```
-
+        
         Then you can install ai-toolkit
         ```bash
         git clone https://github.com/ostris/ai-toolkit.git
@@ -311,12 +300,17 @@ with gr.Blocks(theme=theme, css=css) as demo:
         pip3 install -r requirements.txt
         cd ..
         ```
+
+        Login with Hugging Face to access FLUX.1 [dev], choose a token with `write` permissions to push your LoRAs to the HF Hub
+        ```bash
+        huggingface-cli login
+        ```
         
         Now you can run FLUX LoRA Ease locally by doing a simple 
         ```py
         python app.py
         ```
-        If you prefer command line, you can run Ostris' [AI Toolkit](https://github.com/ostris/ai-toolkit) yourself.
+        If you prefer command line, you can run Ostris' [AI Toolkit](https://github.com/ostris/ai-toolkit) yourself directly.
         ''')
         
     dataset_folder = gr.State()