Add auth token for diffusion model download.

app.py

@@ -21,6 +21,9 @@ from diffusers import DiffusionPipeline, EulerAncestralDiscreteScheduler
 # https://docs.fast.ai/learner.html#load_learner
 from transformers import GPT2TokenizerFast
 
+import os
+AUTH_TOKEN = os.environ.get('AUTH_TOKEN')
+
 # update requirements.txt with:
 # C:\Users\Grant\PycharmProjects\test_space\venv\Scripts\pip3.exe freeze > requirements.txt
 
@@ -59,7 +62,7 @@ class DropOutput(Callback):
 #     is_mega=True,
 #     is_reusable=True
 # )
-pipeline = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", custom_pipeline="stable_diffusion_mega", torch_dtype=torch.float16, revision="fp16")
+pipeline = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", custom_pipeline="stable_diffusion_mega", torch_dtype=torch.float16, revision="fp16", use_auth_token=AUTH_TOKEN)
 
 # pipeline.to("cuda")