Update app.py

app.py

@@ -9,10 +9,16 @@ import re
 
 def generate_image(pipe, prompt, seed=42, randomize_seed=True, width=768, height=768, guidance_scale=4.5, num_inference_steps=20):
     MAX_SEED = np.iinfo(np.int32).max
+    
+    # 无论如何都先创建 generator
+    generator = torch.Generator(device=pipe.device)
+
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
-        generator = torch.Generator(device=pipe.device).manual_seed(seed)
-        print(f"ℹ️  使用种子: {seed}")
+    
+    # 使用最终确定的种子来设置 generator
+    generator.manual_seed(seed)
+    print(f"ℹ️  使用种子: {seed}")
     
     print("🚀 开始生成图像...")
     image = pipe(