Update README.md

README.md

@@ -1,3 +1,39 @@
 ---
-license: openrail++
+language: 
+  - en
+thumbnail: "https://huggingface.co/nousr/robo-diffusion/resolve/main/robo_example.png"
+tags:
+- robots
+- stable-diffusion
+- aiart
+- text-to-image
+license: "openrail++"
 ---
+# Robo-Diffusion 2 (base) - *Alpha Release*
+A dreambooth-method finetune of stable diffusion that will output cool looking robots when prompted.
+
+<img src="https://huggingface.co/nousr/robo-diffusion-2-base/resolve/main/example_grid.png"/>
+
+# Usage
+Keep the words `nousr robot` towards the beginning of your prompt to invoke the finetuned style. Use negative prompts to achieve the best result.
+
+```python
+import torch
+from diffusers import StableDiffusionPipeline, EulerDiscreteScheduler
+
+scheduler = EulerDiscreteScheduler.from_pretrained(MODEL_ID, subfolder="scheduler")
+pipe = StableDiffusionPipeline.from_pretrained("nousr/robo-diffusion-2-base", scheduler=scheduler, torch_dtype=torch.float16, revision="fp16")
+pipe = pipe.to("cuda")
+
+prompt = "A realistic photograph of a 3d nousr robot in a modern city. A glossy white and orange nousr robot."
+negative_prompt = "black and white robot, picture frame, a children's drawing in crayon. #Wholesale, Abstract Metal Sculpture. i'm leaving a bad review."
+image = pipe(prompt, negative_prompt=negative_prompt, num_inference_steps=32, guidance_scale=5.0).images[0]
+
+image.save("robo.png")
+```
+
+# Original Model
+Based on stable diffusion 1.4 can be found [here](https://huggingface.co/nousr/robo-diffusion)
+
+---
+*NOTE: ensure you have read the license and agree to the terms