Update app.py

app.py

@@ -59,12 +59,9 @@ original_pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-
 
 def merge_and_run(prompt, negative_prompt, shuffled_items, lora_1_scale=0.5, lora_2_scale=0.5, progress=gr.Progress(track_tqdm=True)):
   state_dict_1 = copy.deepcopy(shuffled_items[0]['state_dict'])
-  print("state_1", state_dict_1)
   state_dict_2 = copy.deepcopy(shuffled_items[1]['state_dict'])
-  print("state_2", state_dict_2)
   pipe = copy.deepcopy(original_pipe)
-  pipe.to("cuda")  
-    
+  pipe.to("cuda") 
   
   pipe.load_lora_weights(state_dict_1)
   pipe.fuse_lora(lora_1_scale)
@@ -75,10 +72,10 @@ def merge_and_run(prompt, negative_prompt, shuffled_items, lora_1_scale=0.5, lor
     negative_prompt = None
       
   image = pipe(prompt=prompt, negative_prompt=negative_prompt, num_inference_steps=20, width=768, height=768).images[0]
-  yield image
   del pipe
   gc.collect()
   torch.cuda.empty_cache()
+  return image
 
 def get_description(item):
       trigger_word = item["trigger_word"]