Update app.py

app.py

@@ -69,10 +69,9 @@ hf_token = os.environ.get("HF_TOKEN")
 
 # Load the transformer with the token
 dit = Flux2Transformer2DModel.from_pretrained(
-    repo_id,
+    "black-forest-labs/FLUX.1-dev",
     subfolder="transformer",
-    torch_dtype=torch.bfloat16,
-    token=hf_token  # <--- CRITICAL ADDITION
+    token=os.environ.get("HF_TOKEN"),
 )
 
 # Load the pipeline with the token