Update app.py

app.py

@@ -138,9 +138,11 @@ def generate_image(prompt,
     print("Beign to generate")
     image_list = []
     for i in range(img_num):
-        generator = torch.Generator(device="cuda")
         seed = int(randomize_seed_fn(seeds, random_seed))
-        generator = torch.Generator().manual_seed(seed)
+        if torch.cuda.is_available():
+            generator = torch.Generator('cuda').manual_seed(seed)
+        else:
+            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,