Update app.py

app.py

@@ -14,7 +14,7 @@ else:
 # Load the Stable Diffusion 3.5 model
 model_id = "stabilityai/stable-diffusion-3.5-medium"
 pipe = StableDiffusion3Pipeline.from_pretrained(model_id, torch_dtype=torch.float16)
-pipe.to("cuda")
+pipe.to("cpu")
 
 # Define the path to the LoRA model
 lora_model_path = "https://huggingface.co/spaces/DonImages/Testing2/resolve/main/lora_model.pth"  # LoRA model path
@@ -22,7 +22,7 @@ lora_model_path = "https://huggingface.co/spaces/DonImages/Testing2/resolve/main
 # Custom method to load and apply LoRA weights to the Stable Diffusion pipeline
 def load_lora_model(pipe, lora_model_path):
     # Load the LoRA weights (assuming it's a PyTorch .pth file)
-    lora_weights = torch.load(lora_model_path, map_location="cuda")
+    lora_weights = torch.load(lora_model_path, map_location="cpu")
 
     # Modify this section based on how LoRA is intended to interact with your Stable Diffusion model
     # Here, we just load the weights into the model's parameters (this is a conceptual approach)