Update app.py

app.py

@@ -51,7 +51,7 @@ def generate_image(prompt):
         guidance_scale=3.5,
         num_inference_steps=50,
         max_sequence_length=512,
-        generator=torch.Generator("cpu").manual_seed(0)
+        generator=torch.Generator("gpu").manual_seed(0)
 ).images[0]
     except Exception as e:
         # Optionally, handle errors (you can also return a default error image)