Update app.py

app.py

@@ -22,7 +22,7 @@ def train_dreambooth_blora_sdxl(instance_data_dir, b_lora_trained_folder, instan
         "--pretrained_model_name_or_path=stabilityai/stable-diffusion-xl-base-1.0",
         f"--instance_data_dir={instance_data_dir}",
         f"--output_dir={b_lora_trained_folder}",
-        f"--instance_prompt={instance_prompt}",
+        f"--instance_prompt='{instance_prompt}'",
         "--resolution=1024",
         "--rank=64",
         "--train_batch_size=1",
@@ -47,8 +47,6 @@ 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