Update utils.py

utils.py

@@ -146,4 +146,8 @@ def generate_gradio_images(prompt, num_inference_steps, loss_flag = False):
     for seed_no, sd in zip(seed_list, sdconcepts):
         prompts = [f'{prompt} {sd}']
         latents = generate_latents(prompts,num_inference_steps, seed_no, loss_apply=loss_flag)
-        latents_list.append(latents)
+        latents_list.append(latents)
+    # show all
+    latents_list = torch.vstack(latents_list)
+    images = latents_to_pil(latents_list)
+    return images