Update app.py

app.py

@@ -51,7 +51,6 @@ class calculateDuration:
 def generate_images(prompt, trigger_word, steps, seed, cfg_scale, width, height, lora_scale, num_images, progress):
     generator = torch.Generator(device=device).manual_seed(seed)
     images = []
-    seeds = []
     
     with calculateDuration("Generating images"):
         for _ in range(num_images):
@@ -66,9 +65,7 @@ def generate_images(prompt, trigger_word, steps, seed, cfg_scale, width, height,
                 joint_attention_kwargs={"scale": lora_scale},
             ).images[0]
             images.append(image)
-            seeds.append(seed)
-    
-    return images, seeds
+    return images
 
 def run_lora(prompt, cfg_scale, steps, selected_repo, randomize_seed, seed, width, height, lora_scale, num_images, progress=gr.Progress(track_tqdm=True)):
     if not selected_repo:
@@ -93,20 +90,16 @@ def run_lora(prompt, cfg_scale, steps, selected_repo, randomize_seed, seed, widt
         if randomize_seed:
             seed = random.randint(0, MAX_SEED)
     
-    images, seeds = generate_images(prompt, trigger_word, steps, seed, cfg_scale, width, height, lora_scale, num_images, progress)
+    images = generate_images(prompt, trigger_word, steps, seed, cfg_scale, width, height, lora_scale, num_images, progress)
     pipe.to("cpu")
     pipe.unload_lora_weights()
-    return images, seeds  
+    return images, seed  
 
 def update_selection(evt: gr.SelectData):
     index = evt.index
     selected_lora = loras[index]
     return f"Selected LoRA: {selected_lora['title']}", selected_lora["repo"]
 
-def display_seed(evt: gr.SelectData, seeds):
-    index = evt.index
-    return seeds[index]
-
 run_lora.zerogpu = True
 
 css = '''
@@ -157,7 +150,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as app:
             )
         with gr.Column(scale=1):  # Generated images on the right
             result = gr.Gallery(label="Generated Images")
-            seed_display = gr.Number(label="Seed of Selected Image", value=0, interactive=False)
+            seed = gr.Number(label="Seed", value=0, interactive=False)
 
     with gr.Column():
         with gr.Row():
@@ -180,18 +173,12 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as app:
         outputs=[selected_lora_text, selected_repo]
     )
 
-    result.select(
-        fn=display_seed,
-        inputs=[result, gr.State()],
-        outputs=[seed_display]
-    )
-
     generate_button = gr.Button("Generate", variant="primary", elem_id="gen_btn")
     generate_button.click(
         run_lora,
         inputs=[prompt, cfg_scale, steps, selected_repo, randomize_seed, seed, width, height, lora_scale, num_images],
-        outputs=[result, seed_display]
+        outputs=[result, seed]
     )
 
 app.queue()
-app.launch()
+app.launch()