Update app.py

app.py

@@ -14,7 +14,7 @@ with open(lora_list, "r") as file:
     data = json.load(file)
     sdxl_loras = [
         {
-            "image": item["image"],
+            "image": item["image"] if item["image"].startswith("https://") else f"https://huggingface.co/spaces/multimodalart/LoraTheExplorer/resolve/main/{item["image"]}",
             "title": item["title"],
             "repo": item["repo"],
             "trigger_word": item["trigger_word"],
@@ -46,16 +46,19 @@ original_pipe = copy.deepcopy(pipe)
 
 #@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)):
+  print("Copying pipe")
   pipe = copy.deepcopy(original_pipe)
+  print("Loading LoRAs")
   pipe.load_lora_weights(shuffled_items[0]['repo'], weight_name=shuffled_items[0]['weights'])
   pipe.fuse_lora(lora_1_scale)
   pipe.load_lora_weights(shuffled_items[1]['repo'], weight_name=shuffled_items[1]['weights'])
   pipe.fuse_lora(lora_2_scale)
-
+  
   pipe.to(torch_dtype=torch.float16)
   pipe.to("cuda")
   if negative_prompt == "":
     negative_prompt = False
+  print("Running inference")
   image = pipe(prompt=prompt, negative_prompt=negative_prompt, num_inference_steps=25, guidance_scale=7).images[0]
   return image
 
@@ -86,12 +89,12 @@ with gr.Blocks(css=css) as demo:
   )
   with gr.Row():
     with gr.Column(min_width=10, scale=6):
-      lora_1 = gr.Image(interactive=False, height=350)
+      lora_1 = gr.Image(interactive=False, height=300)
       lora_1_prompt = gr.Markdown()
     with gr.Column(min_width=10, scale=1, elem_id="plus_column"):
       plus = gr.HTML("+", elem_id="plus_button")
     with gr.Column(min_width=10, scale=6):
-      lora_2 = gr.Image(interactive=False, height=350)
+      lora_2 = gr.Image(interactive=False, height=300)
       lora_2_prompt = gr.Markdown()
   with gr.Row():
     prompt = gr.Textbox(label="Your prompt", info="arrange the trigger words of the two LoRAs in a coherent sentence", interactive=True, elem_id="prompt")