Update app.py

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-
+import os
 import torch
 from torch import autocast
 from diffusers import StableDiffusionPipeline
@@ -22,7 +22,7 @@ pipe = StableDiffusionPipeline.from_pretrained(
 pipe = pipe.to(device)
 # When running locally, you won`t have access to this, so you can remove this part
 word_list_dataset = load_dataset(
-    "stabilityai/word-list", data_files="list.txt", use_auth_token=True)
+    "stabilityai/word-list", data_files="list.txt", use_auth_token=os.environ.get('auth_token')
 word_list = word_list_dataset["train"]['text']