Update app.py

app.py

@@ -36,9 +36,9 @@ def train_dreambooth_blora_sdxl(instance_data_dir, b_lora_trained_folder, instan
         f"--instance_data_dir={instance_data_dir}",
         f"--output_dir={b_lora_trained_folder}",
         f"--instance_prompt='{instance_prompt}'",
-        f"--validation_prompt=a teddy bear in {instance_prompt} style",
+        #f"--validation_prompt=a teddy bear in {instance_prompt} style",
         "--num_validation_images=1",
-        "--validation_epochs=300",
+        "--validation_epochs=500",
         "--resolution=1024",
         "--rank=64",
         "--train_batch_size=1",
@@ -87,8 +87,8 @@ def main(image_path, b_lora_trained_folder, instance_prompt):
     shutil.copy(image_path, local_dir)
     print(f"source image has been copied in {local_dir} directory")
     
-    max_train_steps = 1000
-    checkpoint_steps = 500
+    max_train_steps = 2000
+    checkpoint_steps = 1000
     
     train_dreambooth_blora_sdxl(local_dir, b_lora_trained_folder, instance_prompt, max_train_steps, checkpoint_steps)