Update app.py

app.py

@@ -53,8 +53,8 @@ If you have any questions, please feel free to open an issue or directly reach u
 
 
 model_num_of_layers = {
-    'Stable Diffusion 1.4': 12,
-    'Stable Diffusion 2.1': 22,
+    'Stable Diffusion 1.4': 13,
+    'Stable Diffusion 2.1': 23,
 }
 
 # def run_for_examples(prompt, model, seed, skip):
@@ -69,7 +69,7 @@ def generate_images(prompt, model, seed, skip):
     max_num_of_layers = model_num_of_layers[model]
     start_layer = max_num_of_layers % skip
     
-    for skip_layers in range(max_num_of_layers - start_layer - 1, -1, -1 * skip):
+    for skip_layers in range(max_num_of_layers - start_layer, -1, -1 * skip):
         # Pass the model and seed to the get_images function
         images = get_images(prompt, skip_layers=skip_layers, model=model, seed=seed)
         all_images.append((images[0], f'layer_{max_num_of_layers - skip_layers}'))