Update app.py

app.py

@@ -10,7 +10,7 @@ def generate_images(prompt):
     all_images = []  # Initialize a list to store all images
     for skip_layers in range(12, -1, -1):
         images = get_images(prompt, skip_layers=skip_layers)
-        all_images.append((images[0], f'layer_{12 - skip_layers}'))  # Add the new image to the list
+        all_images.append(images[0]) # (images[0], f'layer_{12 - skip_layers}'))  # Add the new image to the list
         yield all_images  # Yield the list of all images
 
 with gr.Blocks() as demo: