Update app.py

app.py

@@ -1,3 +1,4 @@
+import os
 import gradio as gr
 
 import torch
@@ -12,8 +13,10 @@ model_id = "runwayml/stable-diffusion-v1-5"
 SAVED_CHECKPOINT = 'mikegarts/distilgpt2-lotr'
 MIN_WORDS = 120
 
+READ_TOKEN = os.environ.get('HF_ACCESS_TOKEN', None)
+
 def get_image_pipe():
-    pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16, revision="fp16")
+    pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16, revision="fp16", use_auth_token=READ_TOKEN)
     pipe.to(pipe.device)
     return pipe