Update app.py

app.py

@@ -13,7 +13,7 @@ def set_accelerate_default_config():
     except subprocess.CalledProcessError as e:
         print(f"An error occurred: {e}")
 
-def train_dreambooth_lora_sdxl(instance_data_dir, lora-trained-xl-folder, instance_prompt, max_train_steps, checkpoint_steps):
+def train_dreambooth_lora_sdxl(instance_data_dir, lora_trained_xl_folder, instance_prompt, max_train_steps, checkpoint_steps):
     
     script_filename = "train_dreambooth_lora_sdxl.py"  # Assuming it's in the same folder
 
@@ -24,7 +24,7 @@ def train_dreambooth_lora_sdxl(instance_data_dir, lora-trained-xl-folder, instan
         "--pretrained_model_name_or_path=stabilityai/stable-diffusion-xl-base-1.0",
         "--pretrained_vae_model_name_or_path=madebyollin/sdxl-vae-fp16-fix",
         f"--instance_data_dir={instance_data_dir}",
-        f"--output_dir={lora-trained-xl-folder}",
+        f"--output_dir={lora_trained_xl_folder}",
         "--mixed_precision=fp16",
         f"--instance_prompt={instance_prompt}",
         "--resolution=1024",
@@ -81,7 +81,7 @@ def main(dataset_id,
     gr.Info("Training begins ...")
 
     instance_data_dir = repo_parts[-1]
-    train_dreambooth_lora_sdxl(instance_data_dir, lora-trained-xl-folder, instance_prompt, max_train_steps, checkpoint_steps)
+    train_dreambooth_lora_sdxl(instance_data_dir, lora_trained_xl_folder, instance_prompt, max_train_steps, checkpoint_steps)
 
     return f"Done, your trained model has been stored in your models library: your_user_name/{lora-trained-xl-folder}"
 
@@ -100,8 +100,8 @@ with gr.Blocks() as demo:
         fn = main,
         inputs = [
             dataset_id,
-            instance_prompt,
             model_output_folder,
+            instance_prompt,
             max_train_steps,
             checkpoint_steps
         ],