Update app.py

app.py

@@ -45,7 +45,7 @@ pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
 lora_repo = "strangerzonehf/Flux-Super-Realism-LoRA"
 trigger_word = "Super Realism"  # Leave blank if no trigger word is needed.
 pipe.load_lora_weights(lora_repo)
-pipe.to("cuda")
+pipe.enable_model_cpu_offload()  # Enable CPU offload to manage GPU memory efficiently
 
 # Define style prompts for Flux.1
 style_list = [