Upload app.py

app.py

@@ -6,7 +6,7 @@ from diffusers import DiffusionPipeline, AutoencoderTiny, AutoencoderKL, AutoPip
 from live_preview_helpers import calculate_shift, retrieve_timesteps, flux_pipe_call_that_returns_an_iterable_of_images
 from diffusers.utils import load_image
 from diffusers import FluxControlNetPipeline, FluxControlNetModel, FluxMultiControlNetModel, FluxControlNetImg2ImgPipeline
-from huggingface_hub import hf_hub_download, HfFileSystem, ModelCard, snapshot_download
+from huggingface_hub import hf_hub_download, HfFileSystem, ModelCard, snapshot_download, HfApi
 import os
 import copy
 import random
@@ -117,6 +117,15 @@ def download_file_mod(url, directory=os.getcwd()):
     if not path: raise Exception(f"Download error: {url}")
     return path
 
+def is_repo_public(repo_id: str):
+    api = HfApi()
+    try:
+        if api.repo_exists(repo_id=repo_id, token=False): return True
+        else: return False
+    except Exception as e:
+        print(f"Error: Failed to connect {repo_id}. {e}")
+        return False
+
 class calculateDuration:
     def __init__(self, activity_name=""):
         self.activity_name = activity_name
@@ -245,6 +254,7 @@ 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
             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: