Update utils.py

utils.py

@@ -140,15 +140,9 @@ def generate_latents(prompts, num_inference_steps, seed_nums, loss_apply=False):
 def pil_to_np(image):
     return np.array(image)
     
-def generate_gradio_images(prompt, num_inference_steps):
+def generate_gradio_images(prompt, num_inference_steps, loss_flag = False):
     # after loss is applied
-    latents_list = [] 
-    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)
-    loss_flag = True
+    latents_list = []
     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)