Update app.py

app.py

@@ -6,12 +6,11 @@ import torch
 
 SEED = 42
 AUTH_TOKEN = os.environ.get("auth_token")
-HF_TOKEN = os.environ.get("HF_TOKEN")
 DEVICE = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
 pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", use_auth_token=AUTH_TOKEN)
 pipe = pipe.to(DEVICE)
-hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "celebrity-set-dataset")
+hf_writer = gr.HuggingFaceDatasetSaver(AUTH_TOKEN, "celebrity-set-dataset")
 
 # Ensure consistently generated images
 generator = torch.Generator(device=DEVICE).manual_seed(SEED)