Update app.py

app.py

@@ -134,9 +134,7 @@ def shuffle_images():
     prompt = gr.update(value=f"{trigger_word_1} {trigger_word_2}")
     scale = gr.update(value=0.7)
 
-    
-    
-    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
+    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
 
 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"