Update app.py

app.py

@@ -71,7 +71,7 @@ css = '''
 #prompt input{width: calc(100% - 160px);border-top-right-radius: 0px;border-bottom-right-radius: 0px;}
 #run_button{position:absolute;margin-top: 36px;right: 0;margin-right: 1.5em;border-bottom-left-radius: 0px;
     border-top-left-radius: 0px;}
-.random_column{align-self: center; align-items: center;gap: 0.5em}
+.random_column{align-self: center; align-items: center;gap: 0.5em !important}
 #share-btn-container{padding-left: 0.5rem !important; padding-right: 0.5rem !important; background-color: #000000; justify-content: center; align-items: center; border-radius: 9999px !important; max-width: 13rem; margin-left: auto;margin-top: 0.35em;}
 div#share-btn-container > div {flex-direction: row;background: black;align-items: center}
 #share-btn-container:hover {background-color: #060606}
@@ -94,8 +94,8 @@ def merge_and_run(prompt, negative_prompt, shuffled_items, lora_1_scale=0.5, lor
   repo_id_1 = shuffled_items[0]['repo']
   repo_id_2 = shuffled_items[1]['repo']
   print("Loading state dicts...")
-  state_dict_1 = copy.deepcopy(state_dicts[repo_id_1]["state_dict"].to("cuda"))
-  state_dict_2 = copy.deepcopy(state_dicts[repo_id_2]["state_dict"].to("cuda"))
+  state_dict_1 = copy.deepcopy(state_dicts[repo_id_1]["state_dict"])
+  state_dict_2 = copy.deepcopy(state_dicts[repo_id_2]["state_dict"])
   print("Loaded state dicts.")
   pipe = copy.deepcopy(original_pipe)
   pipe.to("cuda")