hf hub download

README.md

@@ -1,16 +1,12 @@
----
 title: Interior AI Designer
-emoji: 👀
-colorFrom: blue
-colorTo: pink
+emoji: 🏠
+colorFrom: green
+colorTo: blue
 sdk: gradio
 sdk_version: 4.37.2
 app_file: app.py
 header: mini
 theme: bethecloud/storj_theme
-pinned: false
+pinned: true
 license: apache-2.0
-short_description: Ikea could never
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+short_description: Ikea could never

app.py

@@ -21,7 +21,7 @@ from diffusers import (
     StableDiffusionControlNetPipeline,
     # AutoencoderKL,
 )
-from huggingface_hub import cached_download, hf_hub_url
+from huggingface_hub import hf_hub_download, hf_hub_url
 from controlnet_aux_local import NormalBaeDetector
 
 MAX_SEED = np.iinfo(np.int32).max
@@ -56,7 +56,7 @@ class Preprocessor:
             return
         elif name == "NormalBae":
             print("Loading NormalBae")
-            model_file = cached_download(hf_hub_url(self.MODEL_ID, filename="NormalBaeDetector.pth"))
+            model_file = hf_hub_download(self.MODEL_ID, filename="NormalBaeDetector.pth")
             self.model = NormalBaeDetector.from_pretrained(model_file).to("cuda")
             torch.cuda.empty_cache()
             self.name = name
@@ -81,7 +81,7 @@ torch.cuda.max_memory_allocated(device="cuda")
 # Controlnet Normal
 model_id = "lllyasviel/control_v11p_sd15_normalbae"
 print("initializing controlnet")
-controlnet_file = cached_download(hf_hub_url(model_id, filename="diffusion_pytorch_model.safetensors"))
+controlnet_file = hf_hub_download(model_id, filename="diffusion_pytorch_model.safetensors")
 controlnet = ControlNetModel.from_pretrained(
     controlnet_file,
     torch_dtype=torch.float16,
@@ -105,7 +105,7 @@ controlnet = ControlNetModel.from_pretrained(
 
 # Scheduler
 scheduler_repo = "runwayml/stable-diffusion-v1-5"
-scheduler_file = cached_download(hf_hub_url(scheduler_repo, filename="scheduler/scheduler_config.json", subfolder="scheduler"))
+scheduler_file = hf_hub_download(scheduler_repo, filename="scheduler/scheduler_config.json", subfolder="scheduler")
 scheduler = DPMSolverMultistepScheduler.from_pretrained(
     scheduler_file,
     solver_order=2,
@@ -142,7 +142,7 @@ vae_url = "https://huggingface.co/stabilityai/sd-vae-ft-mse-original/blob/main/v
 
 # Stable Diffusion Pipeline
 base_model_repo = "Lykon/AbsoluteReality"
-base_model_file = cached_download(hf_hub_url(base_model_repo, filename="AbsoluteReality_1.8.1_pruned.safetensors"))
+base_model_file = hf_hub_download(base_model_repo, filename="AbsoluteReality_1.8.1_pruned.safetensors")
 
 print('loading pipe')
 pipe = StableDiffusionControlNetPipeline.from_single_file(