Update app.py

app.py

@@ -10,7 +10,7 @@ HF_TOKEN = os.getenv('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)
-pip = pipe.to(DEVICE)
+pipe = pipe.to(DEVICE)
 hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "celebrity-set-dataset")
 
 # Ensure consistently generated images
@@ -45,7 +45,7 @@ gr.Interface(
             label = 'Movie / Show / Setting')
         ],
     fn = generate,
-    outputs = "image"
+    outputs = "image",
     allow_flagging = "manual",
     flagging_options = ["Looks good", "Looks bad"],
     flagging_callback = hf_writer