Update app.py

app.py

@@ -10,7 +10,7 @@ from time import sleep
 import copy
 import json
 import gc
-import random
+
 with open("sdxl_loras.json", "r") as file:
     data = json.load(file)
     sdxl_loras = [
@@ -23,8 +23,6 @@ with open("sdxl_loras.json", "r") as file:
             "is_compatible": item["is_compatible"],
             "is_pivotal": item.get("is_pivotal", False),
             "text_embedding_weights": item.get("text_embedding_weights", None),
-            "likes": item.get("likes", 0),
-            "downloads": item.get("downloads", 0),
             "is_nc": item.get("is_nc", False)
         }
         for item in data
@@ -43,8 +41,6 @@ for item in sdxl_loras:
     item["saved_name"] = saved_name
     item["state_dict"] = state_dict #{k: v.to(device=device, dtype=torch.float16) for k, v in state_dict.items() if torch.is_tensor(v)}
 
-gr_sdxl_loras = gr.State(value=sdxl_loras)
-
 vae = AutoencoderKL.from_pretrained(
     "madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16
 )
@@ -59,7 +55,7 @@ pipe.to(device)
 last_lora = ""
 last_merged = False
 last_fused = False
-def update_selection(selected_state: gr.SelectData, sdxl_loras):
+def update_selection(selected_state: gr.SelectData):
     lora_repo = sdxl_loras[selected_state.index]["repo"]
     instance_prompt = sdxl_loras[selected_state.index]["trigger_word"]
     new_placeholder = "Type a prompt. This LoRA applies for all prompts, no need for a trigger word" if instance_prompt == "" else "Type a prompt to use your selected LoRA"
@@ -139,7 +135,7 @@ def merge_incompatible_lora(full_path_lora, lora_scale):
                 del lora_model
                 gc.collect()
 
-def run_lora(prompt, negative, lora_scale, selected_state, sdxl_loras, progress=gr.Progress(track_tqdm=True)):
+def run_lora(prompt, negative, lora_scale, selected_state, progress=gr.Progress(track_tqdm=True)):
     global last_lora, last_merged, last_fused, pipe
 
     if negative == "":
@@ -199,17 +195,6 @@ def run_lora(prompt, negative, lora_scale, selected_state, sdxl_loras, progress=
     gc.collect()
     return image, gr.update(visible=True)
 
-def shuffle_gallery(sdxl_loras):
-    random.shuffle(sdxl_loras)
-    return [(item["image"], item["title"]) for item in sdxl_loras], sdxl_loras
-
-def swap_gallery(order, sdxl_loras):
-    if(order == "random"):
-        return shuffle_gallery(sdxl_loras)
-    else:
-        sorted_gallery = sorted(sdxl_loras, key=lambda x: x.get(order, 0), reverse=True)
-        return [(item["image"], item["title"]) for item in sorted_gallery], sdxl_loras
-
 
 with gr.Blocks(css="custom.css") as demo:
     title = gr.HTML(
@@ -218,16 +203,14 @@ with gr.Blocks(css="custom.css") as demo:
     )
     selected_state = gr.State()
     with gr.Row():
-        with gr.Box():
-            order_gallery = gr.Radio(choices=["random", "likes", "downloads"], value="random", label="Order by")
-            gallery = gr.Gallery(
-                value=[(item["image"], item["title"]) for item in sdxl_loras],
-                label="SDXL LoRA Gallery",
-                allow_preview=False,
-                columns=3,
-                elem_id="gallery",
-                show_share_button=False
-            )
+        gallery = gr.Gallery(
+            value=[(item["image"], item["title"]) for item in sdxl_loras],
+            label="SDXL LoRA Gallery",
+            allow_preview=False,
+            columns=3,
+            elem_id="gallery",
+            show_share_button=False
+        )
         with gr.Column():
             prompt_title = gr.Markdown(
                 value="### Click on a LoRA in the gallery to select it",
@@ -285,15 +268,9 @@ with gr.Blocks(css="custom.css") as demo:
                 submit_disclaimer = gr.Markdown(
                     "This is a curated gallery by me, [apolinário (multimodal.art)](https://twitter.com/multimodalart). I'll try to include as many cool LoRAs as they are submitted! You can [duplicate this Space](https://huggingface.co/spaces/multimodalart/LoraTheExplorer?duplicate=true) to use it privately, and add your own LoRAs by editing `sdxl_loras.json` in the Files tab of your private space."
                 )
-    order_gallery.change(
-        fn=swap_gallery,
-        inputs=[order_gallery, gr_sdxl_loras],
-        outputs=[gallery, gr_sdxl_loras],
-        queue=False
-    )
+
     gallery.select(
         update_selection,
-        inputs=[gr_sdxl_loras],
         outputs=[prompt_title, prompt, prompt, selected_state, use_diffusers, use_uis],
         queue=False,
         show_progress=False,
@@ -305,7 +282,7 @@ with gr.Blocks(css="custom.css") as demo:
         show_progress=False
     ).success(
         fn=run_lora,
-        inputs=[prompt, negative, weight, selected_state, gr_sdxl_loras],
+        inputs=[prompt, negative, weight, selected_state],
         outputs=[result, share_group],
     )
     button.click(
@@ -319,6 +296,6 @@ with gr.Blocks(css="custom.css") as demo:
         outputs=[result, share_group],
     )
     share_button.click(None, [], [], _js=share_js)
-    demo.load(fn=shuffle_gallery, inputs=[gr_sdxl_loras], outputs=[gallery, gr_sdxl_loras], queue=False)
+
 demo.queue(max_size=20)
 demo.launch()