fix link and add usage code

README.md

@@ -33,6 +33,8 @@ widget:
 base_model: stabilityai/stable-diffusion-xl-base-1.0
 instance_prompt: factorio
 license: cc0-1.0
+datasets:
+- piebro/factorio-blueprint-visualizations
 ---
 # Factorio
 
@@ -40,15 +42,42 @@ license: cc0-1.0
 
 ## Model description 
 
-Create abstract art in the style of artfully visualized Factorio blueprints. This Lora is trained using this dataset: https://huggingface.co/datasets/piebro/factorio-blueprint-visualizations.
+Create abstract art in the style of artfully visualized Factorio blueprints. This Lora is trained using this dataset: https://huggingface.co/datasets/piebro/factorio-blueprint-visualizations.
 
 ## Trigger words
 
 You should use `factorio` to trigger the image generation.
 
-
 ## Download model
 
 Weights for this model are available in Safetensors format.
 
 [Download](/piebro/factorio-blueprint-visualizations-sdxl-lora/tree/main) them in the Files & versions tab.
+
+## Usage
+
+```python
+import torch
+from diffusers import DiffusionPipeline, AutoencoderKL
+device = "cuda" if torch.cuda.is_available() else "cpu"
+vae = AutoencoderKL.from_pretrained('madebyollin/sdxl-vae-fp16-fix', torch_dtype=torch.float16)
+pipe = DiffusionPipeline.from_pretrained(
+    "stabilityai/stable-diffusion-xl-base-1.0",
+    vae=vae, torch_dtype=torch.float16, variant="fp16",
+    use_safetensors=True
+)
+pipe.to(device)
+# This is where you load your trained weights
+specific_safetensors = "pytorch_lora_weights.safetensors"
+lora_scale = 0.9
+pipe.load_lora_weights(
+    'piebro/factorio-blueprint-visualizations-sdxl-lora', 
+    weight_name = specific_safetensors,
+)
+prompt = "pastel drawing of factorio"
+image = pipe(
+    prompt=prompt, 
+    num_inference_steps=50,
+    cross_attention_kwargs={"scale": lora_scale}
+).images[0]
+```