Pass HF token to Client for ZeroGPU quota

app.py

@@ -1,4 +1,5 @@
 import base64
+import os
 import tempfile
 import urllib.request
 import gradio as gr
@@ -10,6 +11,18 @@ from daggr import FnNode, GradioNode, Graph
 # These tools are free to use and don't consume credits
 
 
+def _hf_token():
+    """HF token so ZeroGPU quota is used (not 'unlogged user'). Prefer env, then huggingface_hub."""
+    token = os.environ.get("HF_TOKEN")
+    if token:
+        return token
+    try:
+        from huggingface_hub import get_token
+        return get_token()
+    except Exception:
+        return None
+
+
 def _url_to_path(url):
     """Download image from URL to a temp file; Daggr needs file paths to display images."""
     if not url or not isinstance(url, str) or not url.startswith("http"):
@@ -100,7 +113,7 @@ def run_upscaler(image, model_selection="4xBHI_dat2_real"):
     except Exception as e:
         raise RuntimeError(f"Upscaler: could not load image from path: {e}") from e
     try:
-        client = Client("Phips/Upscaler")
+        client = Client("Phips/Upscaler", hf_token=_hf_token())
         result = client.predict(image_arg, model_selection, api_name="/upscale_image")
     except Exception as e:
         raise RuntimeError(f"Upscaler API error: {e}") from e
@@ -140,7 +153,7 @@ def run_z_image_turbo(prompt, height=1024, width=1024, seed=42):
     if not prompt or not isinstance(prompt, str) or not prompt.strip():
         return None
     try:
-        client = Client("hf-applications/Z-Image-Turbo")
+        client = Client("hf-applications/Z-Image-Turbo", hf_token=_hf_token())
         result = client.predict(
             prompt=prompt.strip(),
             height=float(height),
@@ -190,7 +203,7 @@ def run_flux_klein(
     if not prompt or not isinstance(prompt, str) or not prompt.strip():
         return None
     try:
-        client = Client("black-forest-labs/FLUX.2-klein-9B")
+        client = Client("black-forest-labs/FLUX.2-klein-9B", hf_token=_hf_token())
         result = client.predict(
             prompt=prompt.strip(),
             input_images=[],