Update app.py

app.py

@@ -1620,7 +1620,7 @@ if __name__ == "__main__":
     args = parser.parse_args()
     
     pipe = AccDiffusionSDXLPipeline.from_pretrained(args.model_ckpt, torch_dtype=torch.float16).to("cuda")
-    generator = torch.Generator(device='cuda')
+    
     
     # GRADIO MODE
 
@@ -1633,6 +1633,7 @@ if __name__ == "__main__":
                                   "n_cross_replace": {"default_": 1.0, "confetti": 0.8},
                                   }
         seed = args.seed
+        generator = torch.Generator(device='cuda')
         generator = generator.manual_seed(seed)
         
         print(f"Prompt: {prompt}")
@@ -1677,7 +1678,7 @@ if __name__ == "__main__":
             gr.HTML("""
             <div style="display:flex;column-gap:4px;">
                 <a href='https://lzhxmu.github.io/accdiffusion/accdiffusion.html'>
-                    <img src='https://img.shields.io/badge/Project-Page-blue'>
+                    <img src='https://img.shields.io/badge/Project-Page-green'>
                 </a>
                 <a href='https://github.com/lzhxmu/AccDiffusion'>
                     <img src='https://img.shields.io/badge/Code-github-blue'>
@@ -1696,7 +1697,7 @@ if __name__ == "__main__":
             gr.Examples(
                 examples = [
                     ["Astronaut on Mars during sunset."],
-                    ["A fow peeking out from behind a bush."],
+                    ["A fox peeking out from behind a bush."],
                     ["A cute corgi on the lawn."],
                 ],
                 inputs = [prompt]