Update app.py

app.py

@@ -14,7 +14,7 @@ else:
 # Load the Stable Diffusion 3.5 model with lower precision (float16)
 model_id = "stabilityai/stable-diffusion-3.5-large"
 pipe = StableDiffusion3Pipeline.from_pretrained(model_id, torch_dtype=torch.float16)  # Use float16 precision
-pipe.to("cpu")  # Ensuring it runs on CPU
+pipe.to("cuda")  # Ensuring it runs on GPU
 
 # Define the path to the LoRA model
 lora_model_path = "./lora_model.pth"  # Assuming the file is saved locally