Update app.py

app.py

@@ -24,6 +24,8 @@ def train_dreambooth_blora_sdxl(instance_data_dir, b_lora_trained_folder, instan
         f"--output_dir={b_lora_trained_folder}",
         f"--instance_prompt='{instance_prompt}'",
         f"--validation_prompt=a teddy bear in {instance_prompt} style",
+        "--num_validation_images=1",
+        "--validation_epochs=300",
         "--resolution=1024",
         "--rank=64",
         "--train_batch_size=1",
@@ -57,8 +59,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 = 10
-    checkpoint_steps = 10
+    max_train_steps = 1000
+    checkpoint_steps = 500
     
     train_dreambooth_blora_sdxl(local_dir, b_lora_trained_folder, instance_prompt, max_train_steps, checkpoint_steps)