Update app.py

app.py

@@ -59,8 +59,13 @@ 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)):
   print(shuffled_items)
+  state_dict_1 = copy.deepcopy(shuffled_items[0]['state_dict'])
+  state_dict_2 = copy.deepcopy(shuffled_items[1]['state_dict'])
+  
   pipe = copy.deepcopy(original_pipe)
-  pipe.to("cuda")
+  pipe.to("cuda")  
+    
+  
   pipe.load_lora_weights(shuffled_items[0]['state_dict'])
   pipe.fuse_lora(lora_1_scale)
   pipe.load_lora_weights(shuffled_items[1]['state_dict'])