Upload 2 files

app.py

@@ -22,7 +22,7 @@ from mod import (clear_cache, get_repo_safetensors, is_repo_name, is_repo_exists
 from flux import (search_civitai_lora, select_civitai_lora, search_civitai_lora_json,
                   download_my_lora, get_all_lora_tupled_list, apply_lora_prompt,
                   update_loras)
-from modutils import get_t2i_model_info, get_download_file
+from modutils import get_t2i_model_info, download_hf_file
 from tagger.tagger import predict_tags_wd, compose_prompt_to_copy
 from tagger.fl2flux import predict_tags_fl2_flux
 
@@ -113,7 +113,7 @@ def change_base_model(repo_id: str, cn_on: bool, disable_model_cache: bool, prog
 change_base_model.zerogpu = True
 
 def download_file_mod(url, directory=os.getcwd()):
-    path = get_download_file(directory, url)
+    path = download_hf_file(directory, url)
     if not path: raise Exception(f"Download error: {url}")
     return path
 
@@ -254,7 +254,13 @@ 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)
-            image = image if is_repo_public(repo) else None
+            if "http" in image and not is_repo_public(repo):
+                try:
+                    image = download_file_mod(image)
+                except Exception as e:
+                    print(e)
+                    image = None
+            #image = image if is_repo_public(repo) else None
             print(f"Loaded custom LoRA: {repo}")
             existing_item_index = next((index for (index, item) in enumerate(current_loras) if item['repo'] == repo), None)
             if existing_item_index is None:

modutils.py

@@ -99,10 +99,12 @@ def download_hf_file(directory, url, progress=gr.Progress(track_tqdm=True)):
     repo_id, filename, subfolder, repo_type = split_hf_url(url)
     try:
         print(f"Downloading {url} to {directory}")
-        if subfolder is not None: hf_hub_download(repo_id=repo_id, filename=filename, subfolder=subfolder, repo_type=repo_type, local_dir=directory, token=hf_token)
-        else: hf_hub_download(repo_id=repo_id, filename=filename, repo_type=repo_type, local_dir=directory, token=hf_token)
+        if subfolder is not None: path = hf_hub_download(repo_id=repo_id, filename=filename, subfolder=subfolder, repo_type=repo_type, local_dir=directory, token=hf_token)
+        else: path = hf_hub_download(repo_id=repo_id, filename=filename, repo_type=repo_type, local_dir=directory, token=hf_token)
+        return path
     except Exception as e:
         print(f"Failed to download: {e}")
+        return None
 
 
 def download_things(directory, url, hf_token="", civitai_api_key=""):