Update app.py

app.py

@@ -10,7 +10,7 @@ title = r"""
 """
 
 description = r"""
-<b>Based on the paper <a href='https://arxiv.org/abs/2403.05846' target='_blank'>InstantStyle: Diffusion Lens: Interpreting Text Encoders in Text-to-Image Pipelines</a>.<br>
+<b>A demo for the paper <a href='https://arxiv.org/abs/2403.05846' target='_blank'>Diffusion Lens: Interpreting Text Encoders in Text-to-Image Pipelines</a>.<br>
 """
 
 article = r"""
@@ -25,8 +25,7 @@ If our work is helpful for your research or applications, please cite us via:
   journal={arXiv preprint arXiv:2403.05846},
   year={2024}
 }
-}
-```
+}```
 📧 **Contact**
 <br>
 If you have any questions, please feel free to open an issue or directly reach us out at <b>tok@cs.technuin.ac.il</b>.
@@ -44,7 +43,7 @@ def generate_images(prompt, model, seed):
     gr.Info('Generating images from intermediate layers..')
     all_images = []  # Initialize a list to store all images
     max_num_of_layers = model_num_of_layers[model]
-    for skip_layers in range(max_num_of_layers, -1, -1):
+    for skip_layers in range(max_num_of_layers - 1, -1, -1):
         # 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_{12 - skip_layers}'))
@@ -59,7 +58,7 @@ with gr.Blocks() as demo:
     # model_select = gr.Dropdown(label="Select Model", choices=['sd1', 'sd2'])
     # seed_input = gr.Number(label="Enter Seed", value=0)  # Default seed set to 0
     with gr.Column():
-        gallery = gr.Gallery(label="Generated Images", columns=6, rows=2, object_fit="contain", height="auto")
+        gallery = gr.Gallery(label="Generated Images", columns=4, rows=3, object_fit="contain", height="auto")
     # Update the submit function to include the new inputs
 
     
@@ -68,7 +67,7 @@ with gr.Blocks() as demo:
     with gr.Column():
         prompt = gr.Textbox(
             label="Prompt",
-            value="a cat, masterpiece, best quality, high quality",
+            value="A photo of Steve Jobs",
         )
 
     model = gr.Radio(