fffiloni commited on
Commit
c5dded7
·
1 Parent(s): d0b3ad7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -1,5 +1,8 @@
 
 
1
  import os
2
- os.environ.getattribute("HF_TOKEN_SD")
 
3
  import gradio as gr
4
  import torch
5
  from torch import autocast
@@ -8,8 +11,6 @@ from diffusers import StableDiffusionPipeline
8
 
9
  print("hello sylvain")
10
 
11
- YOUR_TOKEN=HF_TOKEN_SD
12
-
13
  device="cpu"
14
 
15
  pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", use_auth_token=YOUR_TOKEN)
 
1
+ from dotenv import load_dotenv
2
+ load_dotenv()
3
  import os
4
+ YOUR_TOKEN = os.environ.get("HF_TOKEN_SD")
5
+
6
  import gradio as gr
7
  import torch
8
  from torch import autocast
 
11
 
12
  print("hello sylvain")
13
 
 
 
14
  device="cpu"
15
 
16
  pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", use_auth_token=YOUR_TOKEN)