andresgtn commited on
Commit
e93792a
·
1 Parent(s): d95e404

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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-movie-poster-celebrity-swap")
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')