Upload 2 files

app.py

@@ -14,7 +14,7 @@ import time
 import requests
 import pandas as pd
 
-from env import models, num_loras, num_cns
+from env import models, num_loras, num_cns, HF_TOKEN
 from mod import (clear_cache, get_repo_safetensors, is_repo_name, is_repo_exists, get_model_trigger,
                  description_ui, compose_lora_json, is_valid_lora, fuse_loras, save_image, preprocess_i2i_image,
                  get_trigger_word, enhance_prompt, set_control_union_image,
@@ -111,6 +111,15 @@ def change_base_model(repo_id: str, cn_on: bool, disable_model_cache: bool, prog
 
 change_base_model.zerogpu = True
 
+def update_loras_state(current_loras, gallery):
+    global loras
+    loras = current_loras
+    gallery_items = [(item["image"], item["title"]) for item in current_loras]
+    print(current_loras)
+    print(gallery_items)
+    print(gallery)
+    return gr.update(value=gallery_items)
+
 class calculateDuration:
     def __init__(self, activity_name=""):
         self.activity_name = activity_name
@@ -235,7 +244,7 @@ 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 add_custom_lora(custom_lora, selected_indices, current_loras):
+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)
@@ -297,7 +306,7 @@ def add_custom_lora(custom_lora, selected_indices, current_loras):
     else:
         return current_loras, gr.update(), gr.update(), gr.update(), selected_indices, gr.update(), gr.update(), gr.update(), gr.update()
 
-def remove_custom_lora(selected_indices, current_loras):
+def remove_custom_lora(selected_indices, current_loras, gallery):
     if current_loras:
         custom_lora_repo = current_loras[-1]['repo']
         # Remove from loras list
@@ -548,7 +557,7 @@ run_lora.zerogpu = True
 def get_huggingface_safetensors(link):
     split_link = link.split("/")
     if len(split_link) == 2:
-        model_card = ModelCard.load(link)
+        model_card = ModelCard.load(link, token=HF_TOKEN)
         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"]:
@@ -556,7 +565,7 @@ def get_huggingface_safetensors(link):
         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
-        fs = HfFileSystem()
+        fs = HfFileSystem(token=HF_TOKEN)
         safetensors_name = None
         try:
             list_of_files = fs.ls(link, detail=False)
@@ -802,12 +811,12 @@ with gr.Blocks(theme='NoCrypt/miku@>=1.2.2', fill_width=True, css=css, delete_ca
     )
     add_custom_lora_button.click(
         add_custom_lora,
-        inputs=[custom_lora, selected_indices, loras_state],
+        inputs=[custom_lora, selected_indices, loras_state, gallery],
         outputs=[loras_state, gallery, selected_info_1, selected_info_2, selected_indices, lora_scale_1, lora_scale_2, lora_image_1, lora_image_2]
     )
     remove_custom_lora_button.click(
         remove_custom_lora,
-        inputs=[selected_indices, loras_state],
+        inputs=[selected_indices, loras_state, gallery],
         outputs=[loras_state, gallery, selected_info_1, selected_info_2, selected_indices, lora_scale_1, lora_scale_2, lora_image_1, lora_image_2]
     )
     gr.on(

mod.py

@@ -5,7 +5,8 @@ from PIL import Image
 from pathlib import Path
 import gc
 import subprocess
-from env import num_cns, model_trigger
+from env import num_cns, model_trigger, HF_TOKEN
+import os
 
 
 subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
@@ -13,7 +14,6 @@ subprocess.run('pip cache purge', shell=True)
 device = "cuda" if torch.cuda.is_available() else "cpu"
 torch.set_grad_enabled(False)
 
-
 control_images = [None] * num_cns
 control_modes = [-1] * num_cns
 control_scales = [0] * num_cns
@@ -60,7 +60,7 @@ def clear_cache():
 
 def get_repo_safetensors(repo_id: str):
     from huggingface_hub import HfApi
-    api = HfApi()
+    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=[])
         files = api.list_repo_files(repo_id=repo_id)