update app.py

app.py

@@ -5,12 +5,13 @@ import random
 # import spaces #[uncomment to use ZeroGPU]
 from diffusers import DiffusionPipeline, FluxPipeline
 import torch
+import os
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 torch_dtype = torch.bfloat16 if torch.cuda.is_available() else torch.float32
 model_repo = "black-forest-labs/FLUX.1-dev"  # Replace to the model you would like to use
 
-pipe = FluxPipeline.from_pretrained(model_repo, torch_dtype=torch_dtype).to(device)
+pipe = FluxPipeline.from_pretrained(model_repo, torch_dtype=torch_dtype, token=f"{os.environ.get('tkn')}").to(device)
 #pipe.enable_model_cpu_offload()
 
 MAX_SEED = np.iinfo(np.int32).max