Auth token added

app.py

@@ -1,6 +1,7 @@
 import streamlit as st
 import time
 import torch
+import os
 from torch import autocast
 from diffusers import StableDiffusionPipeline
 from datasets import load_dataset
@@ -22,11 +23,13 @@ seed = sl4.number_input("Enter seed", 0, 25000, 47, 1)
 model_id = "CompVis/stable-diffusion-v1-4"
 device = "cuda"
 
+auth_token = os.environ.get("StableDiffusion") or True
+
 pipe = StableDiffusionPipeline.from_pretrained(
-    model_id, use_auth_token=True, revision="fp16", torch_dtype=torch.float16)
+    model_id, use_auth_token=auth_token, revision="fp16", torch_dtype=torch.float16)
 pipe = pipe.to(device)
 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=auth_token)
 word_list = word_list_dataset["train"]['text']