Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -7,6 +7,9 @@ Original file is located at
|
|
7 |
https://colab.research.google.com/drive/19xx6Nu4FeiGj-TzTUFxBf-15IkeuFx_F
|
8 |
"""
|
9 |
|
|
|
|
|
|
|
10 |
import gradio as gr
|
11 |
import torch as th
|
12 |
|
@@ -28,7 +31,7 @@ print(device)
|
|
28 |
# init stable diffusion model
|
29 |
pipe = StableDiffusionPipeline.from_pretrained(
|
30 |
"CompVis/stable-diffusion-v1-4",
|
31 |
-
use_auth_token=
|
32 |
).to(device)
|
33 |
|
34 |
|
|
|
7 |
https://colab.research.google.com/drive/19xx6Nu4FeiGj-TzTUFxBf-15IkeuFx_F
|
8 |
"""
|
9 |
|
10 |
+
import os
|
11 |
+
USER_TOKEN = os.environ.getattribute("USER_TOKEN")
|
12 |
+
|
13 |
import gradio as gr
|
14 |
import torch as th
|
15 |
|
|
|
31 |
# init stable diffusion model
|
32 |
pipe = StableDiffusionPipeline.from_pretrained(
|
33 |
"CompVis/stable-diffusion-v1-4",
|
34 |
+
use_auth_token=USER_TOKEN
|
35 |
).to(device)
|
36 |
|
37 |
|