Update app.py

app.py

@@ -58,17 +58,17 @@ css = '''
 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)
   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.load_lora_weights(shuffled_items[0]['state_dict'])
+  pipe.load_lora_weights(state_dict_1)
   pipe.fuse_lora(lora_1_scale)
-  pipe.load_lora_weights(shuffled_items[1]['state_dict'])
+  pipe.load_lora_weights(state_dict_2)
   pipe.fuse_lora(lora_2_scale)
   
   if negative_prompt == "":