Update app.py

app.py

@@ -7,7 +7,7 @@ from diffusers import FluxPipeline
 pipe = FluxPipeline.from_pretrained(
     "sayakpaul/FLUX.1-merged",
     torch_dtype=torch.bfloat16
-)
+).to("cuda")
 pipe.transformer.to(memory_format=torch.channels_last)
 pipe.transformer = torch.compile(pipe.transformer, mode="max-autotune", fullgraph=True)
 pipe.transformer = autoquant(