Update app.py

app.py

@@ -24,7 +24,7 @@ pipeline = StableDiffusion3Pipeline.from_pretrained(
 ).to(device)
 
 # Load the LoRA trained weights once at the start
-lora_path = "lora_trained_model.safetensors"  # Ensure this file is uploaded in the Space
+lora_path = "lora_trained_model.pt"  # Ensure this file is uploaded in the Space
 if os.path.exists(lora_path):
     try:
         pipeline.load_lora_weights(lora_path)  # Use the correct method for loading LoRA weights