Update app.py
Browse files
app.py
CHANGED
@@ -28,6 +28,10 @@ from torchvision import transforms
|
|
28 |
from tqdm.auto import tqdm
|
29 |
from transformers import CLIPFeatureExtractor, CLIPTextModel, CLIPTokenizer
|
30 |
|
|
|
|
|
|
|
|
|
31 |
pretrained_model_name_or_path = "CompVis/stable-diffusion-v1-4" #@param {type:"string"}
|
32 |
|
33 |
from IPython.display import Markdown
|
|
|
28 |
from tqdm.auto import tqdm
|
29 |
from transformers import CLIPFeatureExtractor, CLIPTextModel, CLIPTokenizer
|
30 |
|
31 |
+
MY_SECRET_TOKEN=os.environ.get('HF_TOKEN_SD')
|
32 |
+
YOUR_TOKEN=MY_SECRET_TOKEN
|
33 |
+
device="cpu"
|
34 |
+
|
35 |
pretrained_model_name_or_path = "CompVis/stable-diffusion-v1-4" #@param {type:"string"}
|
36 |
|
37 |
from IPython.display import Markdown
|