Upload 2 files

app.py

@@ -996,7 +996,7 @@ with gr.Blocks(theme='NoCrypt/miku@>=1.2.2', fill_width=True, css=css, delete_ca
             queue=False,
             trigger_mode="once",
             show_api=False,
-        ).success(get_repo_safetensors, [lora_repo[i]], [lora_weights[i]], queue=False, show_api=False
+        ).success(get_repo_safetensors, [lora_repo[i]], [lora_weights[i], lora_info[i]], queue=False, show_api=False
         ).success(apply_lora_prompt_flux, [lora_info[i]], [lora_trigger[i]], queue=False, show_api=False
         ).success(compose_lora_json, [lora_repo_json, lora_num[i], lora_repo[i], lora_wt[i], lora_weights[i], lora_trigger[i]], [lora_repo_json], queue=False, show_api=False)
         

mod.py

@@ -8,7 +8,7 @@ import subprocess
 import os
 import re
 from translatepy import Translator
-from huggingface_hub import HfApi, hf_hub_download
+from huggingface_hub import HfApi, hf_hub_download, ModelCard
 from env import num_cns, model_trigger, HF_TOKEN, CIVITAI_API_KEY, DOWNLOAD_LORA_LIST, DIRECTORY_LORAS
 from modutils import download_things
 
@@ -72,16 +72,19 @@ def clear_cache():
 def get_repo_safetensors(repo_id: str):
     api = HfApi(token=HF_TOKEN)
     try:
-        if not is_repo_name(repo_id) or not is_repo_exists(repo_id): return gr.update(value="", choices=[])
+        tag = "None"
+        if not is_repo_name(repo_id) or not is_repo_exists(repo_id): return gr.update(value="", choices=[]), tag
         files = api.list_repo_files(repo_id=repo_id)
+        model_card = ModelCard.load(repo_id, token=HF_TOKEN)
+        tag = model_card.data.get("instance_prompt", "")
     except Exception as e:
         print(f"Error: Failed to get {repo_id}'s info.")
         print(e)
         gr.Warning(f"Error: Failed to get {repo_id}'s info.")
-        return gr.update(choices=[])
+        return gr.update(choices=[]), tag
     files = [f for f in files if f.endswith(".safetensors")]
-    if len(files) == 0: return gr.update(value="", choices=[])
-    else: return gr.update(value=files[0], choices=files)
+    if len(files) == 0: return gr.update(value="", choices=[]), "None"
+    else: return gr.update(value=files[0], choices=files), tag
 
 
 def expand2square(pil_img: Image.Image, background_color: tuple=(0, 0, 0)):