Update app.py

app.py

@@ -5,9 +5,12 @@ import random
 # import spaces #[uncomment to use ZeroGPU]
 from diffusers import DiffusionPipeline
 import torch
+import os 
+a=os.getenv('hf_key2')
+login(token=a )
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
-model_repo_id = "stabilityai/sdxl-turbo"  # Replace to the model you would like to use
+model_repo_id = "black-forest-labs/FLUX.1-dev"  # Replace to the model you would like to use
 
 if torch.cuda.is_available():
     torch_dtype = torch.float16