Update app.py

app.py

@@ -74,7 +74,7 @@ def query(lora_id, prompt, steps=28, cfg_scale=3.5, randomize_seed=True, seed=-1
         image_bytes = response.content
         image = Image.open(io.BytesIO(image_bytes))
         print(f'\033[1mGeneration {key} completed!\033[0m ({prompt})')
-        return image, seed
+        return image, seed, seed
     except Exception as e:
         print(f"Error when trying to open the image: {e}")
         return None
@@ -130,6 +130,6 @@ with gr.Blocks(theme='Nymbo/Nymbo_Theme', css=css) as app:
         )
 
         
-        text_button.click(query, inputs=[custom_lora, text_prompt, steps, cfg, randomize_seed, seed, width, height], outputs=[image_output,seed_output])
+        text_button.click(query, inputs=[custom_lora, text_prompt, steps, cfg, randomize_seed, seed, width, height], outputs=[image_output,seed_output, seed])
 
 app.launch(show_api=False, share=False)