Update app.py

app.py

@@ -13,6 +13,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 
 # # notebook_login(token_key)
 
+[email protected]
 if torch.cuda.is_available():
     torch.cuda.max_memory_allocated(device=device)
     pipe = AutoPipelineForText2Image.from_pretrained("Shaleen123/kandinsky_2.5", torch_dtype=torch.float16, variant="fp16", use_safetensors=True)