Upload 3 files

README.md

@@ -4,7 +4,7 @@ emoji: 🏆😻
 colorFrom: red
 colorTo: pink
 sdk: gradio
-sdk_version: 5.5.0
+sdk_version: 5.6.0
 app_file: app.py
 pinned: true
 license: mit

app.py

@@ -133,7 +133,7 @@ def change_base_model(repo_id: str, cn_on: bool, disable_model_cache: bool, mode
 change_base_model.zerogpu = True
 
 def download_file_mod(url, directory=os.getcwd()):
-    path = download_hf_file(directory, url)
+    path = download_hf_file(directory, url, hf_token=HF_TOKEN)
     if not path: raise Exception(f"Download error: {url}")
     return path
 
@@ -270,11 +270,30 @@ def randomize_loras(selected_indices, loras_state):
     random_prompt = random.choice(prompt_values)
     return selected_info_1, selected_info_2, selected_indices, lora_scale_1, lora_scale_2, lora_image_1, lora_image_2, random_prompt
 
+def download_loras_images(loras_json: list[dict]):
+    loras_json = copy.deepcopy(loras_json)
+    for i, lora in enumerate(loras_json):
+        repo = lora.get("repo", None)
+        if repo is None: continue
+        if "title" not in lora.keys() or "trigger_word" not in lora.keys() or "image" not in lora.keys():
+            title, _repo, _path, trigger_word, image_def = check_custom_model(repo)
+            if "title" not in lora.keys(): loras_json[i]["title"] = title
+            if "trigger_word" not in lora.keys(): loras_json[i]["trigger_word"] = trigger_word
+            if "image" not in lora.keys(): lora["image"] = image_def
+        image = lora.get("image", None)
+        try:
+            if not is_repo_public(repo) and image is not None and "http" in image: image = download_file_mod(image)
+            loras_json[i]["image"] = image if image else "/home/user/app/custom.png"
+        except Exception as e:
+            print(e)
+            continue
+    return loras_json
+
 def add_custom_lora(custom_lora, selected_indices, current_loras, gallery):
     if custom_lora:
         try:
             title, repo, path, trigger_word, image = check_custom_model(custom_lora)
-            if "http" in image and not is_repo_public(repo):
+            if image is not None and "http" in image and not is_repo_public(repo):
                 try:
                     image = download_file_mod(image)
                 except Exception as e:
@@ -603,7 +622,8 @@ def get_huggingface_safetensors(link):
         base_model = model_card.data.get("base_model")
         print(f"Base model: {base_model}")
         if base_model not in ["black-forest-labs/FLUX.1-dev", "black-forest-labs/FLUX.1-schnell"]:
-            raise Exception("Not a FLUX LoRA!")
+            #raise Exception("Not a FLUX LoRA!")
+            gr.Warning("Not a FLUX LoRA?")
         image_path = model_card.data.get("widget", [{}])[0].get("output", {}).get("url", None)
         trigger_word = model_card.data.get("instance_prompt", "")
         image_url = f"https://huggingface.co/{link}/resolve/main/{image_path}" if image_path else None
@@ -652,6 +672,8 @@ def update_history(new_image, history):
     history.insert(0, new_image)
     return history
 
+loras = download_loras_images(loras)
+
 css = '''
 #gen_column{align-self: stretch}
 #gen_btn{height: 100%}
@@ -732,7 +754,7 @@ with gr.Blocks(theme='NoCrypt/miku@>=1.2.2', fill_width=True, css=css, delete_ca
             with gr.Column():
                 selected_info = gr.Markdown("")
                 gallery = gr.Gallery([(item["image"], item["title"]) for item in loras], label="LoRA Gallery", allow_preview=False,
-                                     columns=5, elem_id="gallery", show_share_button=False, interactive=False)
+                                     columns=4, elem_id="gallery", show_share_button=False, interactive=False)
                 with gr.Group():
                     with gr.Row(elem_id="custom_lora_structure"):
                         custom_lora = gr.Textbox(label="Custom LoRA", info="LoRA Hugging Face path or *.safetensors public URL", placeholder="multimodalart/vintage-ads-flux", scale=3, min_width=150)