Update app.py

app.py

@@ -23,7 +23,7 @@ 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",
         "--resolution=1024",
         "--rank=64",
         "--train_batch_size=1",
@@ -57,8 +57,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 = 10
+    checkpoint_steps = 0
     
     train_dreambooth_blora_sdxl(local_dir, b_lora_trained_folder, instance_prompt, max_train_steps, checkpoint_steps)