Update README.md

README.md

@@ -16,7 +16,7 @@ inference: true
 should probably proofread and complete it, then remove this comment. -->
 
 
-# LoRA text2image fine-tuning - yashvoladoddi37/yash_kanji_diffusion
+# LoRA text2image fine-tuning - yashvoladoddi37/kanji-diffusion-v1-5
 These are LoRA adaption weights for runwayml/stable-diffusion-v1-5. The weights were fine-tuned on the yashvoladoddi37/kanjienglish dataset. You can find some example images in the following. 
 
 ![img_0](./image_0.png)
@@ -24,20 +24,18 @@ These are LoRA adaption weights for runwayml/stable-diffusion-v1-5. The weights
 ![img_2](./image_2.png)
 ![img_3](./image_3.png)
 
-
-
-## Intended uses & limitations
-
 #### How to use
 
 ```python
-# TODO: add an example code snippet for running this diffusion pipeline
-```
-
-#### Limitations and bias
-
-[TODO: provide examples of latent issues and potential remediations]
-
-## Training details
-
-[TODO: describe the data used to train the model]
+from diffusers import StableDiffusionPipeline
+import torch
+
+model_path = "yashvoladoddi37/kanji-diffusion-v1-5"
+pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16, use_safetensors = True).to("cuda")
+pipe.unet.load_attn_procs(model_path)
+pipe.to("cuda")
+
+prompt = "a Kanji meaning YouTube"
+image = pipe(prompt).images[0]
+image.save("youtube-kanji-v1-4.png")
+```