Update app.py

app.py

@@ -21,7 +21,7 @@ style_lora_base_path = "Shakker-Labs"
 pipe = FluxPipeline.from_pretrained(base_path, torch_dtype=torch.bfloat16)
 transformer = FluxTransformer2DModel.from_pretrained(base_path, subfolder="transformer", torch_dtype=torch.bfloat16)
 pipe.transformer = transformer
-pipe.to(device)
+pipe.to("cuda")
 
 def clear_cache(transformer):
     for name, attn_processor in transformer.attn_processors.items():