Instructions on how to run

README.md

@@ -13,14 +13,27 @@ library_name: diffusers
 A model from [Playground](https://playground.com)
 </center>
 
-# How to Run
+## Run Playground v1
+
+First, install dependencies:
 
-1. Install requirements (`diffusers` and `torch`)
 ```bash
-pip install -r requirements.txt
+pip install diffusers torch accelerate transformers
 ```
 
-2. Run demo script
-```bash
-python demo.py
-```
+Run the model:
+
+```python
+from diffusers import DiffusionPipeline
+import torch
+
+pipe = DiffusionPipeline.from_pretrained("playgroundai/playground-v1")
+
+if torch.cuda.is_available():
+    pipe.to("cuda")
+
+prompt = "An Astronaut in a jungle, photorealistic"
+image  = pipe(prompt=prompt).images[0]
+
+image.save("playground-v1.png")
+```