Update app.py

app.py

@@ -26,7 +26,7 @@ num_inference_steps = 24
 if st.button("Generate Image"):
     with st.spinner("Generating..."):
         # Generate image with the given parameters
-        generator = torch.Generator("cuda").manual_seed(0)
+        generator = torch.Generator(device)
         image = pipe(prompt, negative_prompt=negative_prompt, num_inference_steps=num_inference_steps,
                      guidance_scale=cfg_scale, width=width, height=height, generator=generator).images[0]