Update app.py

app.py

@@ -127,14 +127,14 @@ def shuffle_images():
     description_1, trigger_word_1 = get_description(two_shuffled_items[0])
     description_2, trigger_word_2 = get_description(two_shuffled_items[1])
 
-    lora_1_link = f"[{repo_id_1}](https://huggingface.co/{repo_id_1}) ✨"
-    lora_2_link = f"[{repo_id_2}](https://huggingface.co/{repo_id_2}) ✨"
+    lora_1_link = f"[{two_shuffled_items[0]['repo']}](https://huggingface.co/{two_shuffled_items[0]['repo']}) ✨"
+    lora_2_link = f"[{two_shuffled_items[1]['repo']}](https://huggingface.co/{two_shuffled_items[1]['repo']}) ✨"
     prompt_description_1 = gr.update(value=description_1, visible=True)
     prompt_description_2 = gr.update(value=description_2, visible=True)
     prompt = gr.update(value=f"{trigger_word_1} {trigger_word_2}")
     scale = gr.update(value=0.7)
 
-    return two_shuffled_items[0]['repo'], title_1, prompt_description_1, repo_id_1, two_shuffled_items[1]['repo'], title_2, prompt_description_2, repo_id_2, prompt, two_shuffled_items, scale, scale
+    return lora_1_link, title_1, prompt_description_1, repo_id_1, lora_2_link, title_2, prompt_description_2, repo_id_2, prompt, two_shuffled_items, scale, scale
 
 def save_preferences(lora_1_id, lora_1_scale, lora_2_id, lora_2_scale, prompt, generated_image, thumbs_direction, seed):
     image_path = IMAGE_DATASET_DIR / f"{uuid4()}.png"