Update app.py

app.py

@@ -47,6 +47,8 @@ def train_dreambooth_blora_sdxl(instance_data_dir, b_lora_trained_folder, instan
         print(f"An error occurred: {e}")
 
 def main(image_path, b_lora_trained_folder, instance_prompt):
+
+    instance_prompt = str(instance_prompt)
     
     local_dir = "image_to_train"
     # Check if the directory exists and create it if necessary
@@ -56,8 +58,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 = 10
     
     train_dreambooth_blora_sdxl(local_dir, b_lora_trained_folder, instance_prompt, max_train_steps, checkpoint_steps)