Read HF_TOKEN for gated Ideogram model

app.py

@@ -31,6 +31,7 @@ BnB4BitDiffusersQuantizer.check_quantized_param_shape = _check_quantized_param_s
 MODEL_ID = "ideogram-ai/ideogram-4-nf4"
 AOTI_REPO = "multimodalart/i4-block-aoti"
 MAX_SEED = 2**31 - 1
+HF_TOKEN = os.environ.get("HF_TOKEN")
 
 MODES = {
     "Turbo · 12 steps": dict(num_inference_steps=12, guidance_schedule=(7.0,) * 11 + (3.0,) * 1, mu=0.5, std=1.75),
@@ -95,7 +96,7 @@ def normalize_caption(raw_caption):
 
 
 t = time.perf_counter()
-pipe = Ideogram4Pipeline.from_pretrained(MODEL_ID, torch_dtype=torch.bfloat16)
+pipe = Ideogram4Pipeline.from_pretrained(MODEL_ID, torch_dtype=torch.bfloat16, token=HF_TOKEN)
 pipe.transformer.dequantize()
 pipe.unconditional_transformer.dequantize()
 pipe.to("cuda")