Update options/Banner_Model/Image2Image_2.py

options/Banner_Model/Image2Image_2.py

@@ -1,6 +1,9 @@
 import torch
 from controlnet_aux import LineartDetector
 from diffusers import ControlNetModel,UniPCMultistepScheduler,FluxPipeline
+from huggingface_hub import login
+import os
+login(token=os.getenv("TOKEN"))
 
 device= "cuda" if torch.cuda.is_available() else "cpu"
 print("Using device for I2I_2:", device)