code update

app.py

@@ -255,33 +255,33 @@ def image_generator(prompt = "dog", loss_function=None):
     else:
       loss_function = None
         
-  images_without_loss = []
-  images_with_loss = []
-
-  seed_values = [8,16,50,80,128]
-  height = 512                        # default height of Stable Diffusion
-  width = 512                         # default width of Stable Diffusion
-  num_inference_steps = 1           # Number of denoising steps
-  guidance_scale = 7.5                # Scale for classifier-free guidance
-  num_styles = len(style_files)
-
-  for i in range(num_styles):
+    images_without_loss = []
+    images_with_loss = []
+    
+    seed_values = [8,16,50,80,128]
+    height = 512                        # default height of Stable Diffusion
+    width = 512                         # default width of Stable Diffusion
+    num_inference_steps = 1           # Number of denoising steps
+    guidance_scale = 7.5                # Scale for classifier-free guidance
+    num_styles = len(style_files)
+    
+    for i in range(num_styles):
       this_generated_img_1 = generate_images(prompt,style_num = i,random_seed = seed_values[i],custom_loss_fn = None)
       images_without_loss.append(this_generated_img_1)
       if loss_function:
         this_generated_img_2 = generate_images(prompt,style_num = i,random_seed = seed_values[i],custom_loss_fn = loss_function)
         images_with_loss.append(this_generated_img_2)
-
-  generated_sd_images = []
-  titles = ["Arcane Style", "Birb Style", "Dr Strange Style", "Midjourney Style", "Oil Style"]
-
-  for i in range(len(titles)):
+    
+    generated_sd_images = []
+    titles = ["Arcane Style", "Birb Style", "Dr Strange Style", "Midjourney Style", "Oil Style"]
+    
+    for i in range(len(titles)):
     generated_sd_images.append((images_without_loss[i], titles[i])) 
     if images_with_loss != []:
       generated_sd_images.append((images_with_loss[i], titles[i])) 
-
-
-  return display_images_in_rows(generated_sd_images, titles)
+    
+    
+    return display_images_in_rows(generated_sd_images, titles)
 
 
 # Create a wrapper function for show_misclassified_images()