Update app.py

app.py

@@ -100,9 +100,9 @@ def merge_and_run(prompt, negative_prompt, shuffled_items, lora_1_scale=0.5, lor
   pipe = copy.deepcopy(original_pipe)
   pipe.to("cuda") 
   print("Loading LoRA weights...")
-  pipe.load_lora_weights(state_dict_1.to("cuda"))
+  pipe.load_lora_weights(state_dict_1)
   pipe.fuse_lora(lora_1_scale)
-  pipe.load_lora_weights(state_dict_2.to("cuda"))
+  pipe.load_lora_weights(state_dict_2)
   pipe.fuse_lora(lora_2_scale)
   print("Loaded LoRA weights.")
   if negative_prompt == "":