Update app.py

app.py

@@ -139,10 +139,7 @@ def generate_image(prompt,
     image_list = []
     for i in range(img_num):
         seed = int(randomize_seed_fn(seeds, random_seed))
-        if torch.cuda.is_available():
-            generator = torch.Generator('cuda').manual_seed(seed)
-        else:
-            generator = torch.Generator().manual_seed(seed)
+        generator = torch.Generator().manual_seed(seed)
         positive_prompt, negative_prompt = apply_style(style, prompt, n_prompt)
         image = pipe(positive_prompt, num_inference_steps=steps, 
                 guidance_scale=scale, height=height,