Update app.py

app.py

@@ -8,12 +8,13 @@ import random
 # Initialize the base model and specific LoRA
 base_model = "black-forest-labs/FLUX.1-dev"
 pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
-pipe.to("cuda")
 
 lora_repo = "XLabs-AI/flux-RealismLora"
 trigger_word = ""  # Leave trigger_word blank if not used.
 pipe.load_lora_weights(lora_repo)
 
+pipe.to("cuda")
+
 MAX_SEED = 2**32-1
 
 @spaces.GPU(duration=80)