from diffusers import StableDiffusionPipeline
import torch
import PIL
from IPython.display import display

# Load the base model
pipe = StableDiffusionPipeline.from_pretrained(
    "runwayml/stable-diffusion-v1-5",  # base model
    torch_dtype=torch.float16,
).to("cuda")

# Load your LoRA
pipe.load_lora_weights("Flo444/example-lora-realistic")

# Fuse LoRA (optional: set strength 0.7 or whatever you want)
pipe.fuse_lora(lora_scale=0.7)

# Prompt
prompt = "beautiful realistic landscape, mountains, river"

# Generate
image = pipe(prompt, num_inference_steps=30, guidance_scale=7).images[0]

# Display the image directly in Colab
display(image)