Update app.py

app.py

@@ -5,6 +5,9 @@ from diffusers.utils import load_image
 import torch
 from PIL import Image
 
+from huggingface_hub import login
+login(token=os.environ.get('TOKEN'))
+
 pipe_prior = KandinskyPriorPipeline.from_pretrained(
     "kandinsky-community/kandinsky-2-1-prior", torch_dtype=torch.float16
 )