Update README.md

README.md

@@ -8,7 +8,20 @@ license: creativeml-openrail-m
 inference: true
 
 ---
+import torch
+from torch import autocast
+from diffusers import StableDiffusionPipeline
 
+pipe = StableDiffusionPipeline.from_pretrained(
+    'hakurei/waifu-diffusion',
+    torch_dtype=torch.float32
+).to('cuda')
+
+prompt = "1girl, aqua eyes, baseball cap, blonde hair, closed mouth, earrings, green background, hat, hoop earrings, jewelry, looking at viewer, shirt, short hair, simple background, solo, upper body, yellow shirt"
+with autocast("cuda"):
+    image = pipe(prompt, guidance_scale=6)["sample"][0]  
+    
+image.save("test.png")
 # waifu-diffusion v1.3 - Diffusion for Weebs
 
 waifu-diffusion is a latent text-to-image diffusion model that has been conditioned on high-quality anime images through fine-tuning.