Update app.py

app.py

@@ -67,7 +67,9 @@ def create_image(image_pil,
         style_image = Image.open(image_pil).convert('RGB')
     else:  # If it's already a PIL Image
         style_image = image_pil.convert('RGB')
-    
+
+    device = next(pipe.parameters()).device
+    generator = torch.Generator(device=device).manual_seed(randomize_seed_fn(seed, True))
     
     image = pipe(
             width=1024,
@@ -76,7 +78,7 @@ def create_image(image_pil,
             negative_prompt="lowres, low quality, worst quality",
             num_inference_steps=24, 
             guidance_scale=guidance_scale,
-            generator=torch.Generator("cuda").manual_seed(randomize_seed_fn(seed, True)),  ## For ZeroGPU no device="cpu"
+            generator=generator,  ## For ZeroGPU no device="cpu"
             clip_image=style_image,
             ipadapter_scale=scale,
         ).images[0]