Update app.py

app.py

@@ -3,8 +3,11 @@ import modin.pandas as pd
 import torch
 import numpy as np
 from PIL import Image
-#from datasets import load_dataset
 from diffusers import StableDiffusionImg2ImgPipeline
+from huggingface_hub import login
+import os
+
+login(token=os.environ.get('HF_KEY'))
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 pipe = StableDiffusionImg2ImgPipeline.from_pretrained("stabilityai/stable-diffusion-xl-refiner-0.9", torch_dtype=torch.float16, safety_checker=None) if torch.cuda.is_available() else StableDiffusionImg2ImgPipeline.from_pretrained("stabilityai/stable-diffusion-xl-refiner-0.9", safety_checker=None)