Testing 2 Stable Diffusion

app.py

@@ -15,10 +15,19 @@ from diffusers import (
 # Quiet matplotlib cache warning on Spaces
 os.environ.setdefault("MPLCONFIGDIR", "/tmp/mpl")
 
+# ---- HF auth helper (works for public & private/gated repos) ----
+def _hf_auth():
+    # Add this in Space Settings → Variables → New secret
+    # Name: HUGGINGFACE_HUB_TOKEN   Value: <your HF token>
+    tok = os.getenv("HUGGINGFACE_HUB_TOKEN") or os.getenv("HF_TOKEN")
+    if not tok:
+        return {}
+    return {"token": tok, "use_auth_token": tok}
+
 # ---- base models for the two tabs ----
 BASE_MODELS = {
     "stable-diffusion-v1-5": "runwayml/stable-diffusion-v1-5",
-    "dream":    "Lykon/dreamshaper-8",
+    "dream": "Lykon/dreamshaper-8",
 }
 
 # ControlNet (QR Monster v2 for SD15)
@@ -88,7 +97,12 @@ def _base_scheduler_for(pipe):
 def get_cn():
     global _CN
     if _CN is None:
-        _CN = ControlNetModel.from_pretrained(CN_QRMON, torch_dtype=DTYPE, use_safetensors=True)
+        _CN = ControlNetModel.from_pretrained(
+            CN_QRMON,
+            torch_dtype=DTYPE,
+            use_safetensors=True,
+            **_hf_auth()
+        )
     return _CN
 
 def get_qrmon_txt2img_pipe(model_id: str):
@@ -100,6 +114,7 @@ def get_qrmon_txt2img_pipe(model_id: str):
             safety_checker=None,
             use_safetensors=True,
             low_cpu_mem_usage=True,
+            **_hf_auth()
         )
         _CN_TXT2IMG[model_id] = _base_scheduler_for(pipe)
     return _CN_TXT2IMG[model_id]
@@ -113,6 +128,7 @@ def get_qrmon_img2img_pipe(model_id: str):
             safety_checker=None,
             use_safetensors=True,
             low_cpu_mem_usage=True,
+            **_hf_auth()
         )
         _CN_IMG2IMG[model_id] = _base_scheduler_for(pipe)
     return _CN_IMG2IMG[model_id]
@@ -196,7 +212,7 @@ def qr_txt2img_dream(*args):
 with gr.Blocks() as demo:
     gr.Markdown("# ZeroGPU • Method 1: QR Control (two base models)")
 
-    # ---- Tab 1: stable-diffusion-v1-5 (anime/illustration) ----
+    # ---- Tab 1: stable-diffusion-v1-5 ----
     with gr.Tab("stable-diffusion-v1-5"):
         url1       = gr.Textbox(label="URL/Text", value="http://www.mybirdfire.com")
         s_prompt1  = gr.Textbox(label="Style prompt", value="japanese painting, elegant shrine and torii, distant mount fuji, autumn maple trees, warm sunlight, 1girl in kimono, highly detailed, intricate patterns, anime key visual, dramatic composition")
@@ -219,7 +235,7 @@ with gr.Blocks() as demo:
         low_img1   = gr.Image(label="Low-res (Stage A) preview")
         ctrl_img1  = gr.Image(label="Control QR used")
 
-        gr.Button("Generate with Anything v4.5").click(
+        gr.Button("Generate with stable-diffusion-v1-5").click(
             qr_txt2img_anything,
             [url1, s_prompt1, s_negative1, steps1, cfg1, size1, border1, qr_w1, seed1,
              use_hires1, hires_up1, hires_str1, repair1, feather1],
@@ -257,4 +273,5 @@ with gr.Blocks() as demo:
         )
 
 if __name__ == "__main__":
+    # Private Spaces: no share=True
     demo.queue(max_size=12).launch()