Update app.py

app.py

@@ -68,8 +68,8 @@ def create_image(image_pil,
     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))
+    
+    generator = torch.Generator().manual_seed(randomize_seed_fn(seed, True))
     
     image = pipe(
             width=1024,