Update app.py

app.py

@@ -62,20 +62,28 @@ def remote_text_encoder(prompts):
     return prompt_embeds
 
 # Load model
-repo_id = "black-forest-labs/FLUX.2-dev"
+repo_id = "black-forest-labs/FLUX.1-dev" # Verify if it's 1 or 2 on the official page
 
+# Get the token from your secrets
+hf_token = os.environ.get("HF_TOKEN")
+
+# Load the transformer with the token
 dit = Flux2Transformer2DModel.from_pretrained(
     repo_id,
     subfolder="transformer",
-    torch_dtype=torch.bfloat16
+    torch_dtype=torch.bfloat16,
+    token=hf_token  # <--- CRITICAL ADDITION
 )
 
+# Load the pipeline with the token
 pipe = Flux2Pipeline.from_pretrained(
     repo_id,
     text_encoder=None,
     transformer=dit,
-    torch_dtype=torch.bfloat16
+    torch_dtype=torch.bfloat16,
+    token=hf_token  # <--- CRITICAL ADDITION
 )
+
 pipe.to(device)
 
 # Pull pre-compiled Flux2 Transformer blocks from HF hub