Update app.py

app.py

@@ -27,7 +27,7 @@ pipeline = StableDiffusion3Pipeline.from_pretrained(
 lora_path = "lora_trained_model.safetensors"  # Ensure this file is uploaded in the Space
 if os.path.exists(lora_path):
     try:
-        SD3LoraLoaderMixin.load_lora_into_model(pipeline, lora_path)
+        pipeline.load_lora_weights(lora_path)  # Use the correct method for loading LoRA weights
         print("✅ LoRA weights loaded successfully!")
     except Exception as e:
         print(f"❌ Error loading LoRA: {e}")
@@ -65,3 +65,6 @@ with gr.Blocks() as demo:
     output_image = gr.Image(label="Generated Image")
 
     generate_btn.click(generate_image, inputs=[prompt_input, seed_input], outputs=output_image)
+
+# Launch the Gradio app
+demo.launch()