Update app.py

app.py

@@ -31,6 +31,9 @@ saved_names = [
     hf_hub_download(item["repo"], item["weights"]) for item in sdxl_loras
 ]
 
+for item, saved_name in zip(sdxl_loras, saved_names):
+    item["saved_name"] = saved_name
+
 css = '''
 #title{text-align:center}
 #plus_column{align-self: center}
@@ -41,19 +44,18 @@ css = '''
     border-top-left-radius: 0px;}
 '''
 
-
-
 #@spaces.GPU
 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)):
   pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16) 
+  pipe.to(torch_dtype=torch.float16)
+  pipe.to("cuda")
   print("Loading LoRAs")
-  pipe.load_lora_weights(shuffled_items[0]['repo'], weight_name=shuffled_items[0]['weights'])
+  pipe.load_lora_weights(shuffled_items[0]['saved_name'])
   pipe.fuse_lora(lora_1_scale)
-  pipe.load_lora_weights(shuffled_items[1]['repo'], weight_name=shuffled_items[1]['weights'])
+  pipe.load_lora_weights(shuffled_items[1]['saved_name'])
   pipe.fuse_lora(lora_2_scale)
   
-  pipe.to(torch_dtype=torch.float16)
-  pipe.to("cuda")
+  
   if negative_prompt == "":
     negative_prompt = False
   print("Running inference")