Update README.md

README.md

@@ -19,6 +19,10 @@ These are comparion for this model ran on T4 GPU, compared with segmind/tiny-sd,
 
 ![image](./compare.png)
 
+# Training info
+
+Learing Rate: 
+
 ### THIS IS NOT THE FULL DESCRIPTION FOR THIS MODEL.
 
 ### License
@@ -33,13 +37,14 @@ The CreativeML OpenRAIL License specifies:
 
 Code:
 ```python
-
 import torch
 from diffusers import StableDiffusionPipeline
+user = "A candid portrait of an elderly person with deep wrinkles, silver hair captured in natural sunlight, wearing a highly detailed coarse wool sweater, dust motes dancing in the light, soft natural backlight, realistic shadows, authentic expression, shot on 35mm film, grainy texture, masterpiece, ultra-realistic"
 model_id="HyHorX/realistic-diffusion-v1"
+neg_prompt="makeup, young, smooth skin, doll, plastic, fake, bad proportions, blurry, high contrast, artificial lighting."
 pipe=StableDiffusionPipeline.from_pretrained(model_id,torch_dtype=torch.float16)
 pipe=pipe.to("cuda")
-prompt=input("Prompt here:")
-image=pipe(prompt).images[0]
+prompt=user
+image=pipe(prompt,negative_prompt=neg_prompt).images[0]
 image
 ```