Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 import torch
-from diffusers import StableDiffusionPipeline, ControlNetModel
-from safetensors.torch import load_file
+from diffusers import StableDiffusionPipeline
+from huggingface_hub import hf_hub_download
 
 # Initialize the pipeline with CPU
 model_id = "runwayml/stable-diffusion-v1-5"
@@ -10,9 +10,15 @@ pipe = StableDiffusionPipeline.from_pretrained(
     torch_dtype=torch.float32,
 )
 
-# Load your ControlNet LoRA
+# Load your LoRA weights
 lora_path = "naonauno/40k-half-sd15"
-pipe.load_lora_weights(lora_path)
+pipe.load_lora_weights(
+    lora_path,
+    weight_name="40kHalf.safetensors"
+)
+
+# Set the cross attention processor scale
+pipe.fuse_lora(scale=1.0)
 
 def generate_image(prompt, negative_prompt, guidance_scale, steps):
     with torch.no_grad():