Create Generate.py

Generate.py

@@ -0,0 +1,25 @@
+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)