Update README.md

README.md

@@ -49,11 +49,9 @@ pipeline_tag: text-to-image
 
 # Kanji Diffusion v1-4 Model Card
 
-Kanji Diffusion is a latent text-to-image diffusion model capable of hallucinating Kanji characters given any prompt.
+Kanji Diffusion is a latent text-to-image diffusion model capable of hallucinating Kanji characters given any English prompt.
 
-This weights here are intended to be used with the 🧨 Diffusers library. If you are looking for the weights to be loaded into the CompVis Stable Diffusion codebase, [come here](https://huggingface.co/CompVis/stable-diffusion-v-1-4-original)
-
-## Model Details
+## Fine-tuned Model Details
 - **Developed by:** Yashpreet Voladoddi
 - **Model type:** Diffusion-based text-to-image generation model, fine-tuned on Stable Diffusion v1.4 model.
 
@@ -61,9 +59,8 @@ This weights here are intended to be used with the 🧨 Diffusers library. If yo
 In order to run the pipeline and see how my model generates the kanji characters, follow the code flow below on Colab(on T4 GPU runtime, else it takes a long time to infer each image).
 Make sure you have your Huggingface API KEY / ACCESS TOKEN for this.
 
+```python
 import os
-
-'''python
 from google.colab import drive
 drive.mount('/content/drive')
 os.chdir("/content/drive/MyDrive")
@@ -84,20 +81,14 @@ pipe.to("cuda")
 prompt = "A Kanji meaning baby robot"
 image = pipe(prompt).images[0]
 image.save("baby-robot-kanji-v1-4.png")
-'''
+```
 
 ### Limitations
 
-
-
-
-
 ## Training
 
 **Training Data**
 
-
-
 **Hardware:** Nvidia GTX 1650 4GB vRAM | 8GB RAM and T4 GPU on Colab
 
 **Training Script:**