Update app.py

app.py

@@ -6,7 +6,10 @@ from PIL import Image, PngImagePlugin
 from io import BytesIO
 import torch
 
-pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4")
+# required for stable difussion
+auth_token = os.environ.get("auth_token") #in secret space
+
+pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", use_auth_token=auth_token)
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 pipe.to(device)