Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -9,7 +9,7 @@ import os
|
|
9 |
|
10 |
# required for stable difussion
|
11 |
auth_token = os.environ.get("auth_token") #in secret space
|
12 |
-
hf_writer = gr.HuggingFaceDatasetSaver(auth_token, "flagged-
|
13 |
|
14 |
pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", use_auth_token=auth_token)
|
15 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
|
|
9 |
|
10 |
# required for stable difussion
|
11 |
auth_token = os.environ.get("auth_token") #in secret space
|
12 |
+
hf_writer = gr.HuggingFaceDatasetSaver(auth_token, "flagged-find-the-next-james-bond")
|
13 |
|
14 |
pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", use_auth_token=auth_token)
|
15 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|