Update app.py

app.py

@@ -55,10 +55,10 @@ css = '''
 }
 '''
 
-pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16)
-original_pipe = copy.deepcopy(pipe)
+original_pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16)
 
 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)):
+  print(shuffled_items)
   pipe = copy.deepcopy(original_pipe)
   pipe.to("cuda")
   pipe.load_lora_weights(shuffled_items[0]['state_dict'])
@@ -70,10 +70,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"]