iciromaco commited on
Commit
718d2d0
1 Parent(s): 721dc44

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
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