Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 from diffusion_lens import get_images
+import numpy as np
 
 MAX_SEED = np.iinfo(np.int32).max
 
@@ -38,6 +39,7 @@ model_num_of_layers = {
 }
 
 def generate_images(prompt, model, seed):
+    seed = random.randint(0, MAX_SEED) if seed == -1 else seed
     print('calling diffusion lens with model:', model, 'and seed:', seed)
     gr.Info('Generating images from intermediate layers..')
     all_images = []  # Initialize a list to store all images