Fix: Login before import, use use_auth_token param, add debug logging

app.py

@@ -4,22 +4,28 @@ import os
 import random
 import spaces
 import torch
-from diffusers import Flux2KleinPipeline
 from huggingface_hub import login
 
 # Authenticate with HF token for gated model access
-hf_token = os.environ.get("HF_TOKEN")
-if hf_token:
-    login(token=hf_token)
+HF_TOKEN = os.environ.get("HF_TOKEN")
+if HF_TOKEN:
+    print(f"HF_TOKEN found, logging in...")
+    login(token=HF_TOKEN)
+    print("Login successful")
+else:
+    print("WARNING: No HF_TOKEN found in environment!")
+
+from diffusers import Flux2KleinPipeline
 
 dtype = torch.bfloat16
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
+print(f"Loading model with token: {'Yes' if HF_TOKEN else 'No'}")
 pipe = Flux2KleinPipeline.from_pretrained(
     "black-forest-labs/FLUX.2-klein-9B",
     torch_dtype=dtype,
-    token=hf_token
-).to(device)
+    use_auth_token=HF_TOKEN
+)
 
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 2048