Update app.py

app.py

@@ -1,7 +1,14 @@
 from diffusers import DiffusionPipeline
+import torch
 
-pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-3.5-large-turbo")
-pipe.load_lora_weights("prithivMLmods/SD3.5-Turbo-Realism-2.0-LoRA")
+device = "cuda" if torch.cuda.is_available() else "cpu"
+model_repo_id = "stabilityai/stable-diffusion-3.5-large-turbo"
 
-prompt = [run_button.click, prompt.submit]
-image = pipe(prompt).images[0]
+torch_dtype = torch.bfloat16 if torch.cuda.is_available() else torch.float32
+
+pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype)
+pipe = pipe.to(device)
+
+pipe.load_lora_weights("prithivMLmods/SD3.5-Turbo-Realism-2.0-LoRA", weight_name="SD3.5-Turbo-Realism-2.0-LoRA.safetensors")
+trigger_word = "Turbo Realism"  
+pipe.fuse_lora(lora_scale=1.0)